fix: out of memory error

app.py

@@ -41,7 +41,7 @@ if gr.NO_RELOAD:
     pipeline.unet = torch.compile(pipeline.unet, mode="reduce-overhead", fullgraph=True)
 
 def generate(prompt: str, progress=gr.Progress()):
-    progress((0, 100), desc="Starting..")
+    progress((0, 100), desc="Generating first frame...")
     image = base(
         prompt=prompt,
         num_inference_steps=n_sdxl_steps,
@@ -68,7 +68,7 @@ def generate(prompt: str, progress=gr.Progress()):
         ),
     ).images[0]
     image = to_tensor(image)
-    progress((n_sdxl_steps + 1, total_steps), desc="Generating video...")
+    progress((n_sdxl_steps, total_steps), desc="Generating video...")
     frames: list[Image.Image] = pipeline(
         prompt=prompt,
         image=image,
@@ -76,7 +76,7 @@ def generate(prompt: str, progress=gr.Progress()):
         negative_prompt=negative_prompt,
         guidance_scale=9.0,
         generator=generator,
-        decode_chunk_size=4,
+        decode_chunk_size=2,
         num_frames=64,
     ).frames[0]
     progress((total_steps - 1, total_steps), desc="Finalizing...")