config.json

{
  "model_type": "pi05",
  "auto_map": {
    "AutoConfig": "configuration_pi05.Pi05Config",
    "AutoTokenizer": "tokenizer_pi05.PaligemmaTokenizer"
  },
  "pi05": true,
  "action_dim": 32,
  "action_horizon": 10,
  "paligemma_variant": "gemma_2b",
  "action_expert_variant": "gemma_300m",
  "dtype": "bfloat16"
}