Adding `safetensors` variant of this model
#1
by
Virt-io
- opened
No description provided.
Virt-io
changed pull request status to
closed
Hi,
Thanks to your PR and comment, we realized that the U-Net folder did not include the following 'fp16' variants (unlike the other folders) and have checked https://huggingface.co/docs/diffusers/using-diffusers/loading#checkpoint-variants
- diffusion_pytorch_model.fp16.bin
- diffusion_pytorch_model.fp16.safetensors
Thank you for your help and time! Now, the 'fp16' variants are available in the U-Net folders of BK-SDM-{Base, Small, Tiny}