nyarunyarunya commited on
Commit
7381712
·
verified ·
1 Parent(s): 9384a5d

Update wan/image2video_mdinfer_oss_stu.py

Browse files
Files changed (1) hide show
  1. wan/image2video_mdinfer_oss_stu.py +1 -2
wan/image2video_mdinfer_oss_stu.py CHANGED
@@ -431,8 +431,7 @@ class WanI2V:
431
  }
432
  latents = latents.unsqueeze(0)
433
 
434
- oss_steps=[2, 6, 14, 28, 44, 56, 66, 74, 79, 84, 87, 90, 93, 94, 95, 96]####oss544Pmed96-16
435
- # oss_steps=[2, 5, 11, 21, 36, 49, 63, 71, 80, 84, 87, 89, 92, 94, 95, 96]####oss544Pmed96-16
436
  latents_oss = infer_OSS(oss_steps, model, latents, context, self.device, model_kwargs=model_kwargs)
437
 
438
  x0_oss = latents_oss
 
431
  }
432
  latents = latents.unsqueeze(0)
433
 
434
+ oss_steps=[2, 6, 18, 36, 47, 56, 65, 72, 78, 83, 87, 90, 92, 94, 95, 96]
 
435
  latents_oss = infer_OSS(oss_steps, model, latents, context, self.device, model_kwargs=model_kwargs)
436
 
437
  x0_oss = latents_oss