Add validated standalone four-step 2000-step EMA and demo

fc7eaf0 verified about 22 hours ago

301 Bytes

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)

	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)