| 162 | return gr.Dropdown() |
| 163 | |
| 164 | def update_motion_module_2(self, motion_module_dropdown): |
| 165 | self.selected_motion_module = motion_module_dropdown |
| 166 | motion_module_dropdown = os.path.join(self.motion_module_dir, motion_module_dropdown) |
| 167 | motion_module_state_dict = torch.load(motion_module_dropdown, map_location="cpu") |
| 168 | _, unexpected = self.unet_model.load_state_dict(motion_module_state_dict, strict=False) |
| 169 | assert len(unexpected) == 0 |
| 170 | return gr.Dropdown() |
| 171 | |
| 172 | # @spaces.GPU(duration=300) |
| 173 | def magictime( |