Spaces:

aakash0563
/

Text-to-Video

Runtime error

aakash0563 commited on Jan 1, 2024

Commit

3eb6ac8

1 Parent(s): 7bded4c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ device = "cpu"  # Force CPU usage
 # Load pipeline (outside the function for efficiency)
 pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
-pipe.enable_model_cpu_offload()
 pipe.enable_vae_slicing()
 def Generate_video(prompt, video_duration_seconds):

 # Load pipeline (outside the function for efficiency)
 pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+# pipe.enable_model_cpu_offload()
 pipe.enable_vae_slicing()
 def Generate_video(prompt, video_duration_seconds):