Files
AIFSH-ComfyUI-DiffSynth-Studio/diffsynth/models/sdxl_vae_encoder.py
T
2024-06-26 11:35:19 +08:00

16 lines
416 B
Python

from .sd_vae_encoder import SDVAEEncoderStateDictConverter, SDVAEEncoder
class SDXLVAEEncoder(SDVAEEncoder):
def __init__(self):
super().__init__()
self.scaling_factor = 0.13025
def state_dict_converter(self):
return SDXLVAEEncoderStateDictConverter()
class SDXLVAEEncoderStateDictConverter(SDVAEEncoderStateDictConverter):
def __init__(self):
super().__init__()