Update app.py

app.py

@@ -174,7 +174,7 @@ class AnimateController:
                 **OmegaConf.to_container(self.inference_config.noise_scheduler_kwargs))
         ).to("cuda")
 
-        original_state_dict = {k: v.cpu().clone() for k, v in model.state_dict().items()}
+        original_state_dict = {k: v.cpu().clone() for k, v in pipeline.unet.state_dict().items()}
         pipeline.unet = convert_lcm_lora(pipeline.unet, self.lcm_lora_path, spatial_lora_slider)
 
         pipeline.to("cuda")