fix: out of memory error
Browse files
app.py
CHANGED
@@ -76,8 +76,8 @@ def generate(prompt: str, progress=gr.Progress()):
|
|
76 |
negative_prompt=negative_prompt,
|
77 |
guidance_scale=9.0,
|
78 |
generator=generator,
|
79 |
-
decode_chunk_size=
|
80 |
-
num_frames=
|
81 |
).frames[0]
|
82 |
progress((total_steps - 1, total_steps), desc="Finalizing...")
|
83 |
frames = [to_tensor(frame.convert("RGB")).mul(255).byte().permute(1, 2, 0) for frame in frames]
|
|
|
76 |
negative_prompt=negative_prompt,
|
77 |
guidance_scale=9.0,
|
78 |
generator=generator,
|
79 |
+
decode_chunk_size=8,
|
80 |
+
num_frames=64,
|
81 |
).frames[0]
|
82 |
progress((total_steps - 1, total_steps), desc="Finalizing...")
|
83 |
frames = [to_tensor(frame.convert("RGB")).mul(255).byte().permute(1, 2, 0) for frame in frames]
|