(self, device="cuda", torch_dtype=torch.float16)
| 10 | class SD3ImagePipeline(BasePipeline): |
| 11 | |
| 12 | def __init__(self, device="cuda", torch_dtype=torch.float16): |
| 13 | super().__init__(device=device, torch_dtype=torch_dtype, height_division_factor=16, width_division_factor=16) |
| 14 | self.scheduler = FlowMatchScheduler() |
| 15 | self.prompter = SD3Prompter() |
| 16 | # models |
| 17 | self.text_encoder_1: SD3TextEncoder1 = None |
| 18 | self.text_encoder_2: SD3TextEncoder2 = None |
| 19 | self.text_encoder_3: SD3TextEncoder3 = None |
| 20 | self.dit: SD3DiT = None |
| 21 | self.vae_decoder: SD3VAEDecoder = None |
| 22 | self.vae_encoder: SD3VAEEncoder = None |
| 23 | self.model_names = ['text_encoder_1', 'text_encoder_2', 'text_encoder_3', 'dit', 'vae_decoder', 'vae_encoder'] |
| 24 | |
| 25 | |
| 26 | def denoising_model(self): |
nothing calls this directly
no test coverage detected