Update tiny models for PI0ForConditionalGeneration

config.json

@@ -0,0 +1,130 @@
+{
+  "architectures": [
+    "PI0ForConditionalGeneration"
+  ],
+  "chunk_size": 4,
+  "dit_config": {
+    "attention_bias": false,
+    "attention_dropout": 0.0,
+    "bos_token_id": 2,
+    "eos_token_id": 1,
+    "head_dim": 8,
+    "hidden_act": "gelu_pytorch_tanh",
+    "hidden_size": 16,
+    "initializer_range": 0.02,
+    "intermediate_size": 32,
+    "is_causal": true,
+    "max_position_embeddings": 8192,
+    "model_type": "gemma",
+    "num_attention_heads": 2,
+    "num_hidden_layers": 2,
+    "num_key_value_heads": 1,
+    "pad_token_id": 0,
+    "rms_norm_eps": 1e-06,
+    "rope_parameters": {
+      "rope_theta": 10000.0,
+      "rope_type": "default"
+    },
+    "tie_word_embeddings": true,
+    "use_bidirectional_attention": true,
+    "use_cache": true,
+    "vocab_size": 128
+  },
+  "dtype": "float32",
+  "loss_reduction": "mean",
+  "max_action_dim": 8,
+  "max_period": 4.0,
+  "max_state_dim": 8,
+  "min_period": 0.004,
+  "model_type": "pi0",
+  "num_inference_steps": 3,
+  "time_sampling_beta_alpha": 1.5,
+  "time_sampling_beta_beta": 1.0,
+  "time_sampling_offset": 0.001,
+  "time_sampling_scale": 0.999,
+  "transformers_version": "5.16.0.dev0",
+  "vlm_config": {
+    "hidden_size": 2048,
+    "image_token_index": 127,
+    "model_type": "paligemma",
+    "projection_dim": 16,
+    "text_config": {
+      "_name_or_path": "",
+      "architectures": null,
+      "attention_bias": false,
+      "attention_dropout": 0.0,
+      "bos_token_id": 2,
+      "chunk_size_feed_forward": 0,
+      "dtype": null,
+      "eos_token_id": 1,
+      "head_dim": 8,
+      "hidden_act": "gelu_pytorch_tanh",
+      "hidden_size": 16,
+      "id2label": {
+        "0": "LABEL_0",
+        "1": "LABEL_1"
+      },
+      "initializer_range": 0.02,
+      "intermediate_size": 32,
+      "is_encoder_decoder": false,
+      "label2id": {
+        "LABEL_0": 0,
+        "LABEL_1": 1
+      },
+      "max_position_embeddings": 8192,
+      "model_type": "gemma",
+      "num_attention_heads": 2,
+      "num_hidden_layers": 2,
+      "num_image_tokens": 4,
+      "num_key_value_heads": 1,
+      "output_attentions": false,
+      "output_hidden_states": false,
+      "pad_token_id": 0,
+      "problem_type": null,
+      "return_dict": true,
+      "rms_norm_eps": 1e-06,
+      "rope_parameters": {
+        "rope_theta": 10000.0,
+        "rope_type": "default"
+      },
+      "tie_word_embeddings": true,
+      "use_bidirectional_attention": true,
+      "use_cache": true,
+      "vocab_size": 128
+    },
+    "tie_word_embeddings": true,
+    "vision_config": {
+      "_name_or_path": "",
+      "architectures": null,
+      "attention_dropout": 0.0,
+      "chunk_size_feed_forward": 0,
+      "dtype": null,
+      "hidden_act": "gelu_pytorch_tanh",
+      "hidden_size": 16,
+      "id2label": {
+        "0": "LABEL_0",
+        "1": "LABEL_1"
+      },
+      "image_size": 8,
+      "intermediate_size": 32,
+      "is_encoder_decoder": false,
+      "label2id": {
+        "LABEL_0": 0,
+        "LABEL_1": 1
+      },
+      "layer_norm_eps": 1e-06,
+      "model_type": "siglip_vision_model",
+      "num_attention_heads": 2,
+      "num_channels": 3,
+      "num_hidden_layers": 1,
+      "output_attentions": false,
+      "output_hidden_states": false,
+      "patch_size": 4,
+      "problem_type": null,
+      "projection_dim": 16,
+      "return_dict": true,
+      "vision_use_head": false
+    },
+    "vocab_size": 257152
+  }
+}

model.safetensors

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