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

Model on Huggingface #176

Open
daniel-bogdoll opened this issue Oct 17, 2024 · 1 comment
Open

Model on Huggingface #176

daniel-bogdoll opened this issue Oct 17, 2024 · 1 comment

Comments

@daniel-bogdoll
Copy link

Hey,

thanks for the great work! Would you be interested in adding the necessary adjustments/configs to HuggingFace, so that the model can be loaded with the Auto functions from Huggingface transformers? Currently, https://huggingface.co/zongzhuofan/co-detr-vit-large-coco only provides a pth file.

from transformers import AutoImageProcessor, AutoModelForObjectDetection

image_processor = AutoImageProcessor.from_pretrained("zongzhuofan/co-detr-vit-large-coco")
model = AutoModelForObjectDetection.from_pretrained("zongzhuofan/co-detr-vit-large-coco")

One example where this was done is "facebook/detr-resnet-50".

@TempleX98
Copy link
Collaborator

Thanks for your suggestion. However, my focus is currently on other projects, and I don't have much time to develop new features for this project right now. Community PR is welcome.

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

No branches or pull requests

2 participants