kenken999 commited on
Commit
8f00609
1 Parent(s): e614e2d
Files changed (1) hide show
  1. routers/gra_06_video/video.py +2 -2
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.float16,
27
- variant='fp16')
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()