update half precision checkpoint

README.md

@@ -39,7 +39,7 @@ from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 base_model_id = "stabilityai/stable-diffusion-xl-base-1.0"
 repo_name = "tianweiy/DMD2"
-ckpt_name = "dmd2_sdxl_4step_unet.bin"
+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(torch.load(hf_hub_download(repo_name, ckpt_name), map_location="cuda"))
@@ -58,7 +58,7 @@ from huggingface_hub import hf_hub_download
 from safetensors.torch import load_file
 base_model_id = "stabilityai/stable-diffusion-xl-base-1.0"
 repo_name = "tianweiy/DMD2"
-ckpt_name = "dmd2_sdxl_1step_unet.bin"
+ckpt_name = "dmd2_sdxl_1step_unet_fp16.bin"
 # Load model.
 unet = UNet2DConditionModel.from_config(base_model_id, subfolder="unet").to("cuda", torch.float16)
 unet.load_state_dict(torch.load(hf_hub_download(repo_name, ckpt_name), map_location="cuda"))

dmd2_sdxl_1step_unet_fp16.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:facefb60b3e7b1d4ad70f16628fe153672b0749842e2cbe20f8da9f281acf8ba
+size 5135671909

dmd2_sdxl_4step_unet_fp16.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5c2d55d5844bc1387f59ef76d2dcca4557bfce7fb426ac5a34bd23f6372bf9f6
+size 5135671909