Upload PaliGemmaForConditionalGeneration

Paligemma 3B trained on ROCO datasets for 6 epochs

README.md

@@ -1,7 +1,7 @@
 ---
+base_model: google/paligemma-3b-pt-448
 library_name: transformers
 license: gemma
-base_model: google/paligemma-3b-pt-448
 tags:
 - generated_from_trainer
 model-index:

config.json

@@ -1,12 +1,12 @@
 {
-  "_name_or_path": "google/paligemma-3b-pt-448",
+  "_name_or_path": "content/paligemma_rocov2/checkpoint-95000",
+  "_vocab_size": 257216,
   "architectures": [
     "PaliGemmaForConditionalGeneration"
   ],
   "bos_token_id": 2,
   "eos_token_id": 1,
   "hidden_size": 2048,
-  "ignore_index": -100,
   "image_token_index": 257152,
   "model_type": "paligemma",
   "pad_token_id": 0,
@@ -22,8 +22,8 @@
     "torch_dtype": "float32",
     "vocab_size": 257216
   },
-  "torch_dtype": "bfloat16",
-  "transformers_version": "4.44.2",
+  "torch_dtype": "float32",
+  "transformers_version": "4.45.1",
   "vision_config": {
     "hidden_size": 1152,
     "image_size": 448,

generation_config.json

@@ -3,5 +3,5 @@
   "bos_token_id": 2,
   "eos_token_id": 1,
   "pad_token_id": 0,
-  "transformers_version": "4.44.2"
+  "transformers_version": "4.45.1"
 }

model-00001-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4caea4eb1a8ff0de1be4ae08def46c8c6f79a9faa74d333e40fda1f5ddabad8f
+size 4956951424

model-00002-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b68b180520b4ae71341479b03e4cc698f44d057594ec397fc26e3e06e7beb8c5
+size 4999820608

model-00003-of-00003.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2396be16e026cdc11486af3f9a59282916e15c208b0e9740e60a6f39a40dad13
+size 1740714288