fix: out of memory error

app.py

@@ -54,6 +54,7 @@ def generate(prompt: str, progress=gr.Progress()):
             progress=progress,
         ),
     ).images[0]
+    progress((n_sdxl_steps * high_noise_frac, total_steps), desc="Refining first frame...")
     image = refiner(
         prompt=prompt,
         num_inference_steps=n_sdxl_steps,
@@ -75,7 +76,7 @@ def generate(prompt: str, progress=gr.Progress()):
         negative_prompt=negative_prompt,
         guidance_scale=9.0,
         generator=generator,
-        decode_chunk_size=8,
+        decode_chunk_size=4,
         num_frames=64,
     ).frames[0]
     progress((total_steps - 1, total_steps), desc="Finalizing...")

utils.py

@@ -2,6 +2,7 @@ from gradio import Progress
 
 def create_progress_updater(start: int, total: int, desc: str, progress: Progress):
     def updater(pipe, step, timestep, callback_kwargs):
-        progress((step + start + 1, total), desc=desc)
+        if progress:
+            progress((step + start + 1, total), desc=desc)
         return callback_kwargs
     return updater