Update skyreelsinfer/skyreels_video_infer.py

skyreelsinfer/skyreels_video_infer.py

@@ -103,8 +103,8 @@ class SkyReelsVideoSingleGpuInfer:
     def warm_up(self):
         init_kwargs = {
             "prompt": "A woman is dancing in a room",
-            "height": 544,
-            "width": 960,
+            "height": 512,
+            "width": 512,
             "guidance_scale": 6,
             "num_inference_steps": 1,
             "negative_prompt": "Aerial view, aerial view, overexposed, low quality, deformation, a poor composition, bad hands, bad teeth, bad eyes, bad limbs, distortion",
@@ -113,7 +113,7 @@ class SkyReelsVideoSingleGpuInfer:
             "embedded_guidance_scale": 1.0,
         }
         if self.task_type == TaskType.I2V:
-            init_kwargs["image"] = Image.new("RGB", (544, 960), color="black")
+            init_kwargs["image"] = Image.new("RGB", (512, 512), color="black")
         self.pipe(**init_kwargs)
 
     def inference(self, kwargs: Dict[str, Any]):