4-step UNet generation importtorchfromdiffusersimportDiffusionPipeline,UNet2DConditionModel,LCMSchedulerfromhuggingface_hubimporthf_hub_downloadfromsafetensors.torchimportload_filebase_model_id="stabilityai/stable-diffusion-xl-base-1.0"repo_name="tianweiy/DMD2"ckpt_name="dmd2_sdxl_4step_unet_fp16.bin...
float16) base_model_id = "stabilityai/stable-diffusion-xl-base-1.0" repo_name = "tianweiy/DMD2" ckpt_name = "dmd2_sdxl_4step_unet_fp16.bin" # Load model. unet = UNet2DConditionModel.from_config(base_model_id, subfolder="unet").to("cuda", torch.float16) unet.load_state_dict(...