ali-vilab
/

text-to-video-ms-1.7b

TextToVideoSDPipeline

Model card Files Files and versions

patrickvonplaten commited on Mar 22, 2023

Commit

2402636

·

1 Parent(s): 059ee50

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -45,7 +45,7 @@ from diffusers.utils import export_to_video
 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_cpu_model_offload()
 prompt = "Spiderman is surfing"
 video_frames = pipe(prompt, num_inference_steps=25).frames

 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()
 prompt = "Spiderman is surfing"
 video_frames = pipe(prompt, num_inference_steps=25).frames