| from .sd_vae_encoder import SDVAEEncoderStateDictConverter, SDVAEEncoder |
|
|
|
|
| class SDXLVAEEncoder(SDVAEEncoder): |
| def __init__(self, upcast_to_float32=True): |
| super().__init__() |
| self.scaling_factor = 0.13025 |
| |
| @staticmethod |
| def state_dict_converter(): |
| return SDXLVAEEncoderStateDictConverter() |
|
|
|
|
| class SDXLVAEEncoderStateDictConverter(SDVAEEncoderStateDictConverter): |
| def __init__(self): |
| super().__init__() |
|
|
| def from_diffusers(self, state_dict): |
| state_dict = super().from_diffusers(state_dict) |
| return state_dict, {"upcast_to_float32": True} |
| |
| def from_civitai(self, state_dict): |
| state_dict = super().from_civitai(state_dict) |
| return state_dict, {"upcast_to_float32": True} |
|
|
