diff --git a/config.json b/config.json
index 2adcdc6..03ad577 100644
--- a/config.json
+++ b/config.json
@@ -1,6 +1,7 @@
 {
   "_name_or_path": "adirik/owlvit-base-patch32",
   "architectures": [
+    "OwlViTModel",
     "OwlViTForObjectDetection"
   ],
   "initializer_factor": 1.0,