nodes/kijai-ComfyUI-WanVideoWrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'main' into longvie2

Browse Source
This commit is contained in:
kijai
2025-12-29 15:39:43 +02:00
parent b551ec9e31 be41f67fae
commit 55c672028b
2 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
wanvideo/schedulers/ersde_scheduler.py
+1 -1
View File
@@ -34,7 +34,7 @@ class ERSDEScheduler():
sigmas.append(0.0) sigmas.append(0.0)
self.sigmas = torch.FloatTensor(sigmas) self.sigmas = torch.FloatTensor(sigmas)
self.sigmas = self.shift * self.sigmas / (1 + (self.shift - 1) * self.sigmas) self.sigmas = self.shift * self.sigmas / (1 + (self.shift - 1) * self.sigmas)
self.timesteps = self.sigmas * self.num_train_timesteps self.timesteps = self.sigmas[:-1] * self.num_train_timesteps
self.step_index = 0 self.step_index = 0
self.old_denoised = None self.old_denoised = None
self.old_denoised_d = None self.old_denoised_d = None
wanvideo/schedulers/flowmatch_res_multistep.py
+1 -3
View File
@@ -35,9 +35,7 @@ class FlowMatchSchedulerResMultistep():
self.sigmas = torch.FloatTensor(sigmas) self.sigmas = torch.FloatTensor(sigmas)
self.sigmas = self.shift * self.sigmas / \ self.sigmas = self.shift * self.sigmas / \
(1 + (self.shift - 1) * self.sigmas) (1 + (self.shift - 1) * self.sigmas)
self.timesteps = self.sigmas * self.num_train_timesteps self.timesteps = self.sigmas[:-1] * self.num_train_timesteps
#print(f"Timesteps: {self.timesteps}, Sigmas: {self.sigmas}")
def step(self, model_output, timestep, sample): def step(self, model_output, timestep, sample):
if timestep.ndim == 2: if timestep.ndim == 2: