.

options/Video_model/Model.py

@@ -54,28 +54,28 @@ def Video(
     video_path = os.path.join(output_folder, f"{base_count:06d}.mp4")
 
     # Perform computation with appropriate dtype based on device
-    if device == "cuda":
-    # Use float16 for GPU
-        with torch.autocast(device_type='cuda', dtype=torch.float16):
-            frames = pipeline(
-                image, height=height, width=width,
-                num_inference_steps=num_inference_steps,
-                min_guidance_scale=min_guidance_scale,
-                max_guidance_scale=max_guidance_scale,
-                num_frames=num_frames, fps=fps, motion_bucket_id=motion_bucket_id,
-                generator=generator,
-            ).frames[0]
-    else:
+    # if device == "cuda":
+    # # Use float16 for GPU
+    #     with torch.autocast(device_type='cuda', dtype=torch.float16):
+    #         frames = pipeline(
+    #             image, height=height, width=width,
+    #             num_inference_steps=num_inference_steps,
+    #             min_guidance_scale=min_guidance_scale,
+    #             max_guidance_scale=max_guidance_scale,
+    #             num_frames=num_frames, fps=fps, motion_bucket_id=motion_bucket_id,
+    #             generator=generator,
+    #         ).frames[0]
+    # else:
         # Use bfloat16 for CPU as it's supported in torch.autocast
         # with torch.autocast(device_type='cpu', dtype=torch.bfloat16):
-        frames = pipeline(
-            image, height=height, width=width,
-            num_inference_steps=num_inference_steps,
-            min_guidance_scale=min_guidance_scale,
-            max_guidance_scale=max_guidance_scale,
-            num_frames=num_frames, fps=fps, motion_bucket_id=motion_bucket_id,
-            generator=generator,
-        ).frames[0]
+    frames = pipeline(
+        image, height=height, width=width,
+        num_inference_steps=num_inference_steps,
+        min_guidance_scale=min_guidance_scale,
+        max_guidance_scale=max_guidance_scale,
+        num_frames=num_frames, fps=fps, motion_bucket_id=motion_bucket_id,
+        generator=generator,
+    ).frames[0]
 
 
     # Save the generated video