(self, model_manager: ModelManager, prompt_refiner_classes=[])
| 28 | |
| 29 | |
| 30 | def fetch_models(self, model_manager: ModelManager, prompt_refiner_classes=[]): |
| 31 | self.text_encoder_1 = model_manager.fetch_model("sd3_text_encoder_1") |
| 32 | self.text_encoder_2 = model_manager.fetch_model("sd3_text_encoder_2") |
| 33 | self.text_encoder_3 = model_manager.fetch_model("sd3_text_encoder_3") |
| 34 | self.dit = model_manager.fetch_model("sd3_dit") |
| 35 | self.vae_decoder = model_manager.fetch_model("sd3_vae_decoder") |
| 36 | self.vae_encoder = model_manager.fetch_model("sd3_vae_encoder") |
| 37 | self.prompter.fetch_models(self.text_encoder_1, self.text_encoder_2, self.text_encoder_3) |
| 38 | self.prompter.load_prompt_refiners(model_manager, prompt_refiner_classes) |
| 39 | |
| 40 | |
| 41 | @staticmethod |
no test coverage detected