Spaces:
Running
Running
File size: 806 Bytes
d88a4e9 007653c a780cef 007653c a780cef 007653c cb78791 007653c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import torch
from diffusers import StableVideoDiffusionPipeline
# CPU पर लोड करने के लिए
pipeline = StableVideoDiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid")
pipeline.to("cpu", torch.float32) # ✅ GPU की जगह CPU का इस्तेमाल
# इनफरेंस फंक्शन (वीडियो जनरेट करना)
def generate_video(prompt):
video = pipeline(prompt).videos # ✅ अब यह CPU पर रन करेगा
return video
# टेस्टिंग
prompt = "A cute cat playing with a ball"
output_video = generate_video(prompt)
# आउटपुट सेव करें
output_video[0].save("generated_video.mp4") # ✅ CPU पर वीडियो सेव होगा |