#!/usr/bin/env python3
from transformers import CLIPTextModel
from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
from diffusers import AutoencoderKL, UNet2DConditionModel

name = "./unet"
model = UNet2DConditionModel.from_pretrained(name)

model.save_pretrained(name, variant="no_ema")
model.save_pretrained(name, variant="fp16")