| 23 | class FixPipeline(BasePipeline): |
| 24 | |
| 25 | def __init__(self, device="cuda", torch_dtype=torch.float16, tokenizer_path=None): |
| 26 | super().__init__(device=device, torch_dtype=torch_dtype) |
| 27 | self.scheduler = FlowMatchScheduler(shift=5, sigma_min=0.0, extra_one_step=True) |
| 28 | self.prompter = WanPrompter(tokenizer_path=tokenizer_path) |
| 29 | self.text_encoder: WanTextEncoder = None |
| 30 | self.image_encoder: WanImageEncoder = None |
| 31 | self.dit: WanModel = None |
| 32 | self.vae: WanVideoVAE = None |
| 33 | self.model_names = ['text_encoder', 'dit', 'vae'] |
| 34 | self.height_division_factor = 16 |
| 35 | self.width_division_factor = 16 |
| 36 | |
| 37 | |
| 38 | def enable_vram_management(self, num_persistent_param_in_dit=None): |