Spaces:
Build error
Build error
Commit
·
3313be0
1
Parent(s):
c86ce41
Update app.py
Browse files
app.py
CHANGED
|
@@ -113,9 +113,6 @@ class AnimateController:
|
|
| 113 |
|
| 114 |
base_model_dropdown = os.path.join(self.personalized_model_dir, base_model_dropdown)
|
| 115 |
base_model_state_dict = {}
|
| 116 |
-
print(os.getcwd())
|
| 117 |
-
print(base_model_dropdown)
|
| 118 |
-
print(os.path.exists(base_model_dropdown))
|
| 119 |
with safe_open(base_model_dropdown, framework="pt", device="cpu") as f:
|
| 120 |
for key in f.keys(): base_model_state_dict[key] = f.get_tensor(key)
|
| 121 |
|
|
@@ -126,7 +123,6 @@ class AnimateController:
|
|
| 126 |
self.unet.load_state_dict(converted_unet_checkpoint, strict=False)
|
| 127 |
|
| 128 |
self.text_encoder = convert_ldm_clip_checkpoint(base_model_state_dict)
|
| 129 |
-
# return gr.Dropdown.update()
|
| 130 |
|
| 131 |
def update_motion_module(self, motion_module_dropdown):
|
| 132 |
self.selected_motion_module = motion_module_dropdown
|
|
|
|
| 113 |
|
| 114 |
base_model_dropdown = os.path.join(self.personalized_model_dir, base_model_dropdown)
|
| 115 |
base_model_state_dict = {}
|
|
|
|
|
|
|
|
|
|
| 116 |
with safe_open(base_model_dropdown, framework="pt", device="cpu") as f:
|
| 117 |
for key in f.keys(): base_model_state_dict[key] = f.get_tensor(key)
|
| 118 |
|
|
|
|
| 123 |
self.unet.load_state_dict(converted_unet_checkpoint, strict=False)
|
| 124 |
|
| 125 |
self.text_encoder = convert_ldm_clip_checkpoint(base_model_state_dict)
|
|
|
|
| 126 |
|
| 127 |
def update_motion_module(self, motion_module_dropdown):
|
| 128 |
self.selected_motion_module = motion_module_dropdown
|