Spaces:
Running
on
Zero
Running
on
Zero
sd
Browse files
routers/gra_06_video/video.py
CHANGED
@@ -23,8 +23,8 @@ DEFAULT_NUM_FRAMES = min(MAX_NUM_FRAMES,
|
|
23 |
int(os.getenv('DEFAULT_NUM_FRAMES', '16')))
|
24 |
|
25 |
pipe = DiffusionPipeline.from_pretrained('damo-vilab/text-to-video-ms-1.7b',
|
26 |
-
torch_dtype=torch.
|
27 |
-
|
28 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
29 |
pipe.enable_model_cpu_offload()
|
30 |
pipe.enable_vae_slicing()
|
|
|
23 |
int(os.getenv('DEFAULT_NUM_FRAMES', '16')))
|
24 |
|
25 |
pipe = DiffusionPipeline.from_pretrained('damo-vilab/text-to-video-ms-1.7b',
|
26 |
+
torch_dtype=torch.float32) # Use full precision
|
27 |
+
|
28 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
29 |
pipe.enable_model_cpu_offload()
|
30 |
pipe.enable_vae_slicing()
|