|
|
from .sd_vae_decoder import SDVAEDecoder, SDVAEDecoderStateDictConverter |
|
|
|
|
|
|
|
|
class SDXLVAEDecoder(SDVAEDecoder): |
|
|
def __init__(self, upcast_to_float32=True): |
|
|
super().__init__() |
|
|
self.scaling_factor = 0.13025 |
|
|
|
|
|
@staticmethod |
|
|
def state_dict_converter(): |
|
|
return SDXLVAEDecoderStateDictConverter() |
|
|
|
|
|
|
|
|
class SDXLVAEDecoderStateDictConverter(SDVAEDecoderStateDictConverter): |
|
|
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} |
|
|
|
