Spaces:

wangfuyun
/

AnimateLCM

Build error

wangfuyun commited on Feb 4, 2024

Commit

fa0af54

verified ·

1 Parent(s): 823eab2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -196,7 +196,7 @@ class AnimateController:
                 video_length=length_slider,
             ).videos
-        pipeline.unet.load(original_state_dict)
         del original_state_dict
         save_sample_path = os.path.join(

                 video_length=length_slider,
             ).videos
+        pipeline.unet.load_state_dict(original_state_dict)
         del original_state_dict
         save_sample_path = os.path.join(