.

options/Video_model/Model.py

@@ -13,12 +13,15 @@ def Video(image):
 
 
     pipeline = StableVideoDiffusionPipeline.from_pretrained(
-        "stabilityai/stable-video-diffusion-img2vid-xt-1-1", torch_dtype=torch.float32
+        "stabilityai/stable-video-diffusion-img2vid-xt-1-1", torch_dtype=torch.float16
     ).to(device)
 
     # Enable model offloading if using the CPU
     if device == "cpu":
         pipeline.enable_model_cpu_offload()
+    else:
+        pipeline.enable_sequential_cpu_offload()
+
     
     image = Image.fromarray(image)
     image = image.resize((1024, 576))