Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Integrate with huggingface #69

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

Integrate with huggingface #69

wants to merge 2 commits into from

Conversation

not-lain
Copy link

🔥 Summary

This pr introduces better integration with huggingface (and for local usage as well)
allowing users to use from_pretrained push_to_hub, and save_pretrained to directly load and save their models with ease (similar to the transformers library )

📖 📷 Additional context

  • Any new dependencies added? N/A
  • Any breaking changes introduced? backward compatibility is ensured
  • Screenshots?
    image

🛠 Testing instructions

I have prepared this notebook to help you migrate your current checkpoint to the supported format: https://colab.research.google.com/drive/1NdS2sTgJCSP2opLU3rC5BaG_npPDRv0S?usp=sharing

Important

It is essential that you run the notebook above to equip your repo with the new changes

once your repo has been equiped users can directly load your checkpoint with just 2 lines of code

from cube3d.inference.engine import Engine # or EngineFast
model = Engine.from_pretrained("not-lain/cube") 

✅ Checklist

  • Provide testing instructions
  • Update relevant documentation

Do not hesitate if you have any reviews on the pr or any follow ups.

Best regards,
Hafedh Hichri
Huggingface Fellow

@not-lain not-lain requested a review from a team as a code owner March 30, 2025 23:23
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant