BabaK07
/

pixeltext-ai

feature-extraction

vision-language

text-extraction

Model card Files Files and versions

BabaK07 commited on Aug 24, 2025

Commit

aae2a40

·

verified ·

1 Parent(s): b8a8a54

Fix config.json for proper loading

Files changed (1) hide show

config.json +5 -3

config.json CHANGED Viewed

@@ -2,7 +2,7 @@
   "architectures": [
     "FixedPaliGemmaOCR"
   ],
-  "model_type": "custom-paligemma-ocr",
   "base_model": "google/paligemma-3b-pt-224",
   "custom_ocr_features": true,
   "hidden_size": 2048,
@@ -10,6 +10,8 @@
   "torch_dtype": "float32",
   "transformers_version": "4.40.0",
   "auto_map": {
-    "AutoModel": "modeling_paligemma_ocr.FixedPaliGemmaOCR"
-  }
 }

   "architectures": [
     "FixedPaliGemmaOCR"
   ],
+  "model_type": "paligemma",
   "base_model": "google/paligemma-3b-pt-224",
   "custom_ocr_features": true,
   "hidden_size": 2048,
   "torch_dtype": "float32",
   "transformers_version": "4.40.0",
   "auto_map": {
+    "AutoModel": "modeling_pixeltext.FixedPaliGemmaOCR"
+  },
+  "_name_or_path": "google/paligemma-3b-pt-224",
+  "custom_model": true
 }