search

autodistill / autodistill-owl-vit

OWL-ViT module for Autodistill.
https://autodistill.com
Apache License 2.0
5 stars 2 forks source link

Support Custom Models #3

Open djwessel opened 9 months ago

djwessel commented 9 months ago
  1. Moved model initialization into class.
  2. Added support for custom Huggingface checkpoints.
  3. Added proper support of DEVICE.
CLAassistant commented 9 months ago

CLA assistant check
All committers have signed the CLA.

sudoLife commented 6 months ago

@djwessel looks like the repo isn't very active...I submited a PR that fixes the 3rd item on your checklist before seeing this. I guess it would make sense if this PR did the first 2 items only, but let's see what the maintainers think once they check it out.