(self, device="cuda", torch_dtype=torch.float16)
| 14 | class SDImagePipeline(BasePipeline): |
| 15 | |
| 16 | def __init__(self, device="cuda", torch_dtype=torch.float16): |
| 17 | super().__init__(device=device, torch_dtype=torch_dtype) |
| 18 | self.scheduler = EnhancedDDIMScheduler() |
| 19 | self.prompter = SDPrompter() |
| 20 | # models |
| 21 | self.text_encoder: SDTextEncoder = None |
| 22 | self.unet: SDUNet = None |
| 23 | self.vae_decoder: SDVAEDecoder = None |
| 24 | self.vae_encoder: SDVAEEncoder = None |
| 25 | self.controlnet: MultiControlNetManager = None |
| 26 | self.ipadapter_image_encoder: IpAdapterCLIPImageEmbedder = None |
| 27 | self.ipadapter: SDIpAdapter = None |
| 28 | self.model_names = ['text_encoder', 'unet', 'vae_decoder', 'vae_encoder', 'controlnet', 'ipadapter_image_encoder', 'ipadapter'] |
| 29 | |
| 30 | |
| 31 | def denoising_model(self): |
nothing calls this directly
no test coverage detected