safetensor vs checkpoint

#2
by SynysterSocks - opened

Hi, sorry to bother you, and thank you for your great work. on your github implamentation, the readme says "Model weights are available on Google Drive and Hugging Face. We recommend using Google Drive. Please download it and put it inside ./checkpoints/"

is there a reason the model version on googledrive is a .ckpt insted of a safetensor (which is also kindly provided by you on huggingface), im guessing compatability or accidental version upload(googledrive)? thank you.

The reason is that the Transformers library uses the safetensors format by default. So if you call model.push_to_hub("your-hf-username/vfusion3d"), it will push safetensors by default.

It's a bit unclear why the authors recommend Google Drive above the hub, that's... unfortunate :( I assume weights are identical. Maybe @JunlinHan could clarify this

@nielsr I feel this will be updated. I worked on the HF implementation (and converted the weights to safetensors) if you recall, and it was likely a timeline issue (VFusion wasn't on Hugging Face when the GitHub README was written).
Of course safetensors is better!
I can open a PR if @JunlinHan wants.

Thanks, that'd be great!

Yes, here is the PR for this issue on GitHub: https://github.com/facebookresearch/vfusion3d/pull/14
Waiting for Junlin's confirmation!

AI at Meta org

Hey guys, very sorry for my late response.

Many thanks nielsr for raising the issue and providing helpful suggestions!

And thank you Jade for making the PR.

PR accepted, thanks!

JunlinHan changed discussion status to closed

Sign up or log in to comment