patrickvonplaten commited on
Commit
a4db70b
1 Parent(s): 3ccb54b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -33,7 +33,7 @@ This model has a wide range of applications, and can reason and generate videos
33
  Let's first install the libraries required:
34
 
35
  ```bash
36
- $ pip install diffusers transformers git+https://github.com/huggingface/accelerate.git
37
  ```
38
 
39
  Now, generate a video:
@@ -45,7 +45,7 @@ from diffusers.utils import export_to_video
45
 
46
  pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b-legacy", torch_dtype=torch.float16)
47
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
48
- pipe = pipe.to("cuda")
49
 
50
  prompt = "Spiderman is surfing"
51
  video_frames = pipe(prompt, num_inference_steps=25).frames
 
33
  Let's first install the libraries required:
34
 
35
  ```bash
36
+ $ pip install git+https://github.com/huggingface/diffusers transformers accelerate
37
  ```
38
 
39
  Now, generate a video:
 
45
 
46
  pipe = DiffusionPipeline.from_pretrained("damo-vilab/text-to-video-ms-1.7b-legacy", torch_dtype=torch.float16)
47
  pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
48
+ pipe.enable_cpu_model_offload()
49
 
50
  prompt = "Spiderman is surfing"
51
  video_frames = pipe(prompt, num_inference_steps=25).frames