| 20 | |
| 21 | |
| 22 | class TrtUnetOption(sd_unet.SdUnetOption): |
| 23 | def __init__(self, name: str, filename: List[dict]): |
| 24 | self.label = f"[TRT] {name}" |
| 25 | self.model_name = name |
| 26 | self.configs = filename |
| 27 | |
| 28 | def create_unet(self): |
| 29 | return TrtUnet(self.model_name, self.configs) |
| 30 | |
| 31 | |
| 32 | class TrtUnet(sd_unet.SdUnet): |