File size: 301 Bytes
fc7eaf0 | 1 2 3 4 5 6 7 8 9 | import torch
from diffusers import AutoencoderKL
class Flux1Autoencoder(AutoencoderKL):
@classmethod
def from_pretrained(cls, pretrained_model_name_or_path, **kwargs):
kwargs['torch_dtype']=torch.float32
return super().from_pretrained(pretrained_model_name_or_path, **kwargs)
|