Spaces:
Configuration error
Configuration error
naomiKenKorem
commited on
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
|