xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2026-05-04 05:26:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update pipeline_cogvideox_inpaint.py

Browse Source
This commit is contained in:
kijai 2024-10-06 12:58:07 +03:00
parent 502ebab83f
commit 5e2e11dd0a
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
cogvideox_fun/pipeline_cogvideox_inpaint.py
View File

@ -433,6 +433,16 @@ class CogVideoX_Fun_Pipeline_Inpaint(VideoSysPipeline):
extra_step_kwargs["generator"] = generator extra_step_kwargs["generator"] = generator
return extra_step_kwargs return extra_step_kwargs
def _gaussian_weights(self, t_tile_length, t_batch_size):
from numpy import pi, exp, sqrt
var = 0.01
midpoint = (t_tile_length - 1) / 2 # -1 because index goes from 0 to latent_width - 1
t_probs = [exp(-(t-midpoint)*(t-midpoint)/(t_tile_length*t_tile_length)/(2*var)) / sqrt(2*pi*var) for t in range(t_tile_length)]
weights = torch.tensor(t_probs)
weights = weights.unsqueeze(0).unsqueeze(2).unsqueeze(3).unsqueeze(4).repeat(1, t_batch_size,1, 1, 1)
return weights
# Copied from diffusers.pipelines.latte.pipeline_latte.LattePipeline.check_inputs # Copied from diffusers.pipelines.latte.pipeline_latte.LattePipeline.check_inputs
def check_inputs( def check_inputs(
self, self,
@ -869,7 +879,6 @@ class CogVideoX_Fun_Pipeline_Inpaint(VideoSysPipeline):
use_context_schedule = True use_context_schedule = True
from .context import get_context_scheduler from .context import get_context_scheduler
context = get_context_scheduler(context_schedule) context = get_context_scheduler(context_schedule)
else: else:
use_temporal_tiling = False use_temporal_tiling = False
use_context_schedule = False use_context_schedule = False