naomiKenKorem commited on
Commit
3313be0
1 Parent(s): c86ce41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -4
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