From c453bbe4dab0fec6f7df343b09ea87048629c58d Mon Sep 17 00:00:00 2001 From: Dmitry Nechaev <32195253+megavaz@users.noreply.github.com> Date: Wed, 23 Oct 2024 20:17:20 +0500 Subject: [PATCH] Fix README.md Thanks to @kaczmarj for pointing out the wrong info about loading hibou-L --- README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 776c238..de86f3d 100644 --- a/README.md +++ b/README.md @@ -30,10 +30,10 @@ model = AutoModel.from_pretrained("histai/hibou-b", trust_remote_code=True) OR ```python -from transformers import HibouImageProcessor, HibouModel +from transformers import AutoImageProcessor, AutoModel -processor = HibouImageProcessor.from_pretrained("histai/hibou-L", trust_remote_code=True) -model = HibouModel.from_pretrained("histai/hibou-L", trust_remote_code=True) +processor = AutoImageProcessor.from_pretrained("histai/hibou-L", trust_remote_code=True) +model = AutoModel.from_pretrained("histai/hibou-L", trust_remote_code=True) ``` We use a customized implementation of the DINOv2 architecture from the transformers library to add support for registers, which requires the `trust_remote_code=True` flag.