480p, v6e-16

Author: prishajain1

src/maxdiffusion/tests/wan_magcache_test.py

@@ -80,8 +80,8 @@ def setUpClass(cls):
             None,
             os.path.join(THIS_DIR, "..", "configs", "base_wan_14b.yml"),
             "num_inference_steps=50",
-            "height=720",
-            "width=1280",
+            "height=480",
+            "width=832",
             "num_frames=81",
             "seed=11234567893",
             "attention=flash",
@@ -152,8 +152,8 @@ def setUpClass(cls):
             None,
             os.path.join(THIS_DIR, "..", "configs", "base_wan_27b.yml"),
             "num_inference_steps=40",
-            "height=720",
-            "width=1280",
+            "height=480",
+            "width=832",
             "num_frames=81",
             "guidance_scale_low=3.0",
             "guidance_scale_high=4.0",
@@ -233,9 +233,9 @@ def setUpClass(cls):
             None,
             os.path.join(THIS_DIR, "..", "configs", "base_wan_i2v_14b.yml"),
             "num_inference_steps=50",
-            "height=720",
-            "width=1280",
-            "pretrained_model_name_or_path=Wan-AI/Wan2.1-I2V-14B-720P-Diffusers",
+            "height=480",
+            "width=832",
+            "pretrained_model_name_or_path=Wan-AI/Wan2.1-I2V-14B-480P-Diffusers",
             "num_frames=81",
             "seed=11234567893",
             "attention=flash",
@@ -311,8 +311,8 @@ def setUpClass(cls):
             None,
             os.path.join(THIS_DIR, "..", "configs", "base_wan_i2v_27b.yml"),
             "num_inference_steps=40",
-            "height=720",
-            "width=1280",
+            "height=480",
+            "width=832",
             "num_frames=81",
             "guidance_scale_low=3.0",
             "guidance_scale_high=4.0",