Hugging Face's logo

Spaces:
jbilcke-hf
/
Wan2GP_you_must_clone_this_space_to_use_it
Paused

App Files Community
2
DeepBeepMeep commited on
Commit
3c71c3f
·
1 Parent(s): be1972e

Fixe Star zero bug

Browse files
Files changed (2) hide show
  1. wan/image2video.py +1 -1
  2. wan/text2video.py +1 -1
wan/image2video.py CHANGED
@@ -409,7 +409,7 @@ class WanI2V:
409
 
410
 
411
  if (i <= cfg_zero_step):
412
- noise_pred *= 0. # it would be faster not to compute noise_pred...
413
  else:
414
  noise_pred_uncond *= alpha
415
  noise_pred = noise_pred_uncond + guide_scale * (noise_pred_text - noise_pred_uncond)
 
409
 
410
 
411
  if (i <= cfg_zero_step):
412
+ noise_pred = noise_pred_text*0. # it would be faster not to compute noise_pred...
413
  else:
414
  noise_pred_uncond *= alpha
415
  noise_pred = noise_pred_uncond + guide_scale * (noise_pred_text - noise_pred_uncond)
wan/text2video.py CHANGED
@@ -306,7 +306,7 @@ class WanT2V:
306
  alpha = alpha.view(batch_size, 1, 1, 1)
307
 
308
  if (i <= cfg_zero_step):
309
- noise_pred *= 0. # it would be faster not to compute noise_pred...
310
  else:
311
  noise_pred_uncond *= alpha
312
  noise_pred = noise_pred_uncond + guide_scale * (noise_pred_text - noise_pred_uncond)
 
306
  alpha = alpha.view(batch_size, 1, 1, 1)
307
 
308
  if (i <= cfg_zero_step):
309
+ noise_pred = noise_pred_text*0. # it would be faster not to compute noise_pred...
310
  else:
311
  noise_pred_uncond *= alpha
312
  noise_pred = noise_pred_uncond + guide_scale * (noise_pred_text - noise_pred_uncond)