(self, ckpt: str)
| 65 | self.set_unet("None") |
| 66 | |
| 67 | def set_unet(self, ckpt: str): |
| 68 | # TODO test if using this with TRT works |
| 69 | sd_unet.apply_unet(ckpt) |
| 70 | sd_hijack.model_hijack.apply_optimizations(ckpt) |
| 71 | |
| 72 | def get_input_names(self) -> List[str]: |
| 73 | names = ["sample", "timesteps", "encoder_hidden_states"] |
no test coverage detected