patrickvonplaten
commited on
Commit
•
a4db70b
1
Parent(s):
3ccb54b
Update README.md
Browse files
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
|
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
|
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
|