ihsanvp commited on
Commit
6e58b71
1 Parent(s): 344a67a

fix: out of memory error

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=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]
 
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]