(self, model_manager: ModelManager, prompt_refiner_classes=[])
| 24 | |
| 25 | |
| 26 | def fetch_models(self, model_manager: ModelManager, prompt_refiner_classes=[]): |
| 27 | self.text_encoder = model_manager.fetch_model("flux_text_encoder_2") |
| 28 | self.dit = model_manager.fetch_model("cog_dit") |
| 29 | self.vae_encoder = model_manager.fetch_model("cog_vae_encoder") |
| 30 | self.vae_decoder = model_manager.fetch_model("cog_vae_decoder") |
| 31 | self.prompter.fetch_models(self.text_encoder) |
| 32 | self.prompter.load_prompt_refiners(model_manager, prompt_refiner_classes) |
| 33 | |
| 34 | |
| 35 | @staticmethod |
no test coverage detected