fix: out of memory error
Browse files
app.py
CHANGED
@@ -77,7 +77,7 @@ def generate(prompt: str, progress=gr.Progress()):
|
|
77 |
guidance_scale=9.0,
|
78 |
generator=generator,
|
79 |
decode_chunk_size=2,
|
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]
|
|
|
77 |
guidance_scale=9.0,
|
78 |
generator=generator,
|
79 |
decode_chunk_size=2,
|
80 |
+
num_frames=32,
|
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]
|