cpu_2

options/Video_model/Model.py

@@ -6,8 +6,8 @@ from PIL import Image
 device="cuda" if torch.cuda.is_available() else "cpu"
 
 pipeline = StableVideoDiffusionPipeline.from_pretrained(
-    "stabilityai/stable-video-diffusion-img2vid-xt-1-1", torch_dtype=torch.float16, variant="fp16"
-)
+    "stabilityai/stable-video-diffusion-img2vid-xt-1-1", torch_dtype=torch.float32
+).to(device)
 pipeline.enable_model_cpu_offload()
 
 def Video(image):
@@ -15,6 +15,6 @@ def Video(image):
     image = image.resize((1024, 576))
 
     generator = torch.manual_seed(42)
-    frames = pipeline(image, decode_chunk_size=8, generator=generator,device=device).frames[0]
+    frames = pipeline(image, decode_chunk_size=8, generator=generator).frames[0]
     export_to_video(frames, "generated.mp4", fps=7)
     return "generated.mp4"