AnimateLCM

Runtime error

cocktailpeanut commited on Feb 5

Commit

90954f4

•

1 Parent(s): b416df6

update

Files changed (1) hide show

app2.py CHANGED Viewed

@@ -186,7 +186,7 @@ class AnimateController:
         original_state_dict = {k: v.cpu().clone() for k, v in pipeline.unet.state_dict().items() if "motion_modules." not in k}
         pipeline.unet = convert_lcm_lora(pipeline.unet, self.lcm_lora_path, spatial_lora_slider)
-        pipeline.enable_model_cpu_offload()
         pipeline.to(device)
         if seed_textbox != -1 and seed_textbox != "":

         original_state_dict = {k: v.cpu().clone() for k, v in pipeline.unet.state_dict().items() if "motion_modules." not in k}
         pipeline.unet = convert_lcm_lora(pipeline.unet, self.lcm_lora_path, spatial_lora_slider)
+        pipeline.enable_sequential_cpu_offload()
         pipeline.to(device)
         if seed_textbox != -1 and seed_textbox != "":