Update app.py

app.py

@@ -219,7 +219,8 @@ def generate(
         noise_pred_uncond, noise_pred_text = noise_pred.chunk(2)
         noise_pred = noise_pred_uncond + guidance_scale * (noise_pred_text - noise_pred_uncond)
         current_latents = pipe.scheduler.step(noise_pred, t, current_latents, return_dict=False)[0]
-        
+    
+    generator_state = generator.get_state()
     intermediate_latents_cpu = current_latents.detach().cpu()
 
     if segment==8: