Fix README.md

Thanks to @kaczmarj for pointing out the wrong info about loading hibou-L
HistAI · Oct 23, 2024 · c453bbe · c453bbe
1 parent d31873c
commit c453bbe
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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.