BUAADreamer
/

PaliGemma-3B-Chat-v0.2

Image-Text-to-Text

text-generation-inference

Model card Files Files and versions Community

hiyouga commited on Jun 5

Commit

fa0185f

•

1 Parent(s): 6d6b652

Update config.json

Files changed (1) hide show

config.json +2 -3

config.json CHANGED Viewed

@@ -1,5 +1,4 @@
 {
-  "_name_or_path": "saves/paligemma-3b-chat-v0.2",
   "architectures": [
     "PaliGemmaForConditionalGeneration"
   ],
@@ -19,7 +18,7 @@
     "num_hidden_layers": 18,
     "num_image_tokens": 1024,
     "num_key_value_heads": 1,
-    "torch_dtype": "float32",
     "vocab_size": 257216
   },
   "torch_dtype": "bfloat16",
@@ -39,4 +38,4 @@
     "vision_use_head": false
   },
   "vocab_size": 257216
-}

 {
   "architectures": [
     "PaliGemmaForConditionalGeneration"
   ],
     "num_hidden_layers": 18,
     "num_image_tokens": 1024,
     "num_key_value_heads": 1,
+    "torch_dtype": "bfloat16",
     "vocab_size": 257216
   },
   "torch_dtype": "bfloat16",
     "vision_use_head": false
   },
   "vocab_size": 257216
+}