safetensors (without optimizer params) for the model checkpoint?

ali-vilab / AnyDoor

Official implementations for paper: Anydoor: zero-shot object-level image customization

https://ali-vilab.github.io/AnyDoor-Page/

MIT License

4.01k stars 366 forks source link

safetensors (without optimizer params) for the model checkpoint? #90

Open odusseys opened 6 months ago

odusseys commented 6 months ago

Hi, I can't seem to find a download link to the model which does not include the optimizer params. Since this is a huge file, I just cant map it to memory (even loading using CPU is insane), and loading with weights_only does not work Could you consider uploading safetensor weights only, so that we may load the model more reasonably? thanks