Spaces:
Running
on
Zero
Running
on
Zero
Update wan/image2video_mdinfer_oss_stu.py
Browse files
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,
|
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
|