(self)
| 116 | self.motion_module_list = [os.path.basename(p) for p in motion_module_list] |
| 117 | |
| 118 | def refresh_personalized_model(self): |
| 119 | dreambooth_list = glob(os.path.join(self.personalized_model_dir, "*.safetensors")) |
| 120 | self.dreambooth_list = [os.path.basename(p) for p in dreambooth_list] |
| 121 | |
| 122 | def update_dreambooth(self, dreambooth_dropdown, motion_module_dropdown=None): |
| 123 | self.selected_dreambooth = dreambooth_dropdown |