pixeltext-ai / config.json
BabaK07's picture
Optimize config for safetensors loading
4105bb6 verified
{
"architectures": [
"FixedPixelTextOCR"
],
"model_type": "pixeltext",
"base_model": "google/paligemma-3b-pt-224",
"hidden_size": 2048,
"vocab_size": 257216,
"torch_dtype": "float32",
"transformers_version": "4.40.0",
"use_safetensors": true,
"auto_map": {
"AutoModel": "modeling_pixeltext.FixedPixelTextOCR",
"AutoConfig": "modeling_pixeltext.PixelTextConfig"
},
"_name_or_path": "BabaK07/pixeltext-ai",
"custom_model": true
}