ihsanvp commited on
Commit
5a50f18
1 Parent(s): 70afc10

fix: out of memory error

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