.

options/Video_model/Model.py

@@ -3,18 +3,33 @@ from diffusers import StableVideoDiffusionPipeline
 from diffusers.utils import load_image, export_to_video
 from PIL import Image
 
-device="cuda" if torch.cuda.is_available() else "cpu"
+# Check if CUDA (GPU) is available, otherwise use CPU
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
+# Load the pipeline and move it to the appropriate device (GPU or CPU)
 pipeline = StableVideoDiffusionPipeline.from_pretrained(
     "stabilityai/stable-video-diffusion-img2vid-xt-1-1", torch_dtype=torch.float32
 ).to(device)
-pipeline.enable_model_cpu_offload()
 
+# Enable model offloading if using the CPU
+if device == "cpu":
+    pipeline.enable_model_cpu_offload()
+
+# Function to generate the video
 def Video(image):
     image = Image.fromarray(image)
     image = image.resize((1024, 576))
 
+    # Set random seed for reproducibility
     generator = torch.manual_seed(42)
+    
+    # Ensure the image is moved to the appropriate device (GPU or CPU)
+    image = image.to(device)
+    
+    # Generate the video frames
     frames = pipeline(image, decode_chunk_size=8, generator=generator).frames[0]
+    
+    # Export the frames to a video file
     export_to_video(frames, "generated.mp4", fps=7)
+    
     return "generated.mp4"