Commit
•
6934ce5
1
Parent(s):
ea64118
Update README.md
Browse files
README.md
CHANGED
|
@@ -37,6 +37,7 @@ pipe = DiffusionPipeline.from_pretrained("cerspense/zeroscope_v2_576w", torch_dt
|
|
| 37 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
| 38 |
pipe.enable_model_cpu_offload()
|
| 39 |
pipe.enable_vae_slicing()
|
|
|
|
| 40 |
|
| 41 |
prompt = "Darth Vader is surfing on waves"
|
| 42 |
video_frames = pipe(prompt, num_inference_steps=40, height=320, width=576, num_frames=36).frames
|
|
|
|
| 37 |
pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
|
| 38 |
pipe.enable_model_cpu_offload()
|
| 39 |
pipe.enable_vae_slicing()
|
| 40 |
+
pipe.unet.enable_forward_chunking(chunk_size=1, dim=1) # disable if enough memory as this slows down significantly
|
| 41 |
|
| 42 |
prompt = "Darth Vader is surfing on waves"
|
| 43 |
video_frames = pipe(prompt, num_inference_steps=40, height=320, width=576, num_frames=36).frames
|