| 124 | |
| 125 | |
| 126 | def fetch_models(self, model_manager: ModelManager): |
| 127 | text_encoder_model_and_path = model_manager.fetch_model("wan_video_text_encoder", require_model_path=True) |
| 128 | if text_encoder_model_and_path is not None: |
| 129 | self.text_encoder, tokenizer_path = text_encoder_model_and_path |
| 130 | self.prompter.fetch_models(self.text_encoder) |
| 131 | self.prompter.fetch_tokenizer(os.path.join(os.path.dirname(tokenizer_path), "google/umt5-xxl")) |
| 132 | self.dit = model_manager.fetch_model("wan_video_dit") |
| 133 | self.vae = model_manager.fetch_model("wan_video_vae") |
| 134 | self.image_encoder = model_manager.fetch_model("wan_video_image_encoder") |
| 135 | |
| 136 | |
| 137 | @staticmethod |