Spaces:
Running
Running
Commit
•
3eb6ac8
1
Parent(s):
7bded4c
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ device = "cpu" # Force CPU usage
|
|
| 13 |
# Load pipeline (outside the function for efficiency)
|
| 14 |
pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")
|
| 15 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
| 16 |
-
pipe.enable_model_cpu_offload()
|
| 17 |
pipe.enable_vae_slicing()
|
| 18 |
|
| 19 |
def Generate_video(prompt, video_duration_seconds):
|
|
|
|
| 13 |
# Load pipeline (outside the function for efficiency)
|
| 14 |
pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b", torch_dtype=torch.float16, variant="fp16")
|
| 15 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
| 16 |
+
# pipe.enable_model_cpu_offload()
|
| 17 |
pipe.enable_vae_slicing()
|
| 18 |
|
| 19 |
def Generate_video(prompt, video_duration_seconds):
|