Hugging Face's logo

Spaces:
cocktailpeanut
/
AnimateLCM
Runtime error

App Files Community
cocktailpeanut commited on
Commit
b416df6
•
1 Parent(s): 9e85b6d

update

Browse files
Files changed (1) hide show
  1. app2.py +1 -0
app2.py CHANGED
@@ -186,6 +186,7 @@ class AnimateController:
186
  original_state_dict = {k: v.cpu().clone() for k, v in pipeline.unet.state_dict().items() if "motion_modules." not in k}
187
  pipeline.unet = convert_lcm_lora(pipeline.unet, self.lcm_lora_path, spatial_lora_slider)
188
 
 
189
  pipeline.to(device)
190
 
191
  if seed_textbox != -1 and seed_textbox != "":
 
186
  original_state_dict = {k: v.cpu().clone() for k, v in pipeline.unet.state_dict().items() if "motion_modules." not in k}
187
  pipeline.unet = convert_lcm_lora(pipeline.unet, self.lcm_lora_path, spatial_lora_slider)
188
 
189
+ pipeline.enable_model_cpu_offload()
190
  pipeline.to(device)
191
 
192
  if seed_textbox != -1 and seed_textbox != "":