| { |
| "arch_specifier": "linear", |
| "architectures": [ |
| "PrismaticForConditionalGeneration" |
| ], |
| "auto_map": { |
| "AutoConfig": "configuration_prismatic.PrismaticConfig", |
| "AutoModelForVision2Seq": "modelling_pi.PrismaticForConditionalGeneration" |
| }, |
| "hf_llm_id": "agopalkr/gemma-2b", |
| "image_resize_strategy": "resize-naive", |
| "image_sizes": [ |
| 224 |
| ], |
| "llm_backbone_id": "gemma-2b", |
| "llm_max_length": 2048, |
| "model_type": "prismatic", |
| "output_projector_states": false, |
| "pad_to_multiple_of": 64, |
| "pad_token_id": 257153, |
| "text_config": { |
| "bos_token_id": 2, |
| "eos_token_id": 1, |
| "head_dim": 256, |
| "hidden_act": "gelu_pytorch_tanh", |
| "hidden_activation": null, |
| "hidden_size": 2048, |
| "intermediate_size": 24576, |
| "max_position_embeddings": 8192, |
| "model_type": "gemma", |
| "num_attention_heads": 16, |
| "num_hidden_layers": 28, |
| "num_key_value_heads": 16, |
| "pad_token_id": 257153, |
| "tie_word_embeddings": true, |
| "torch_dtype": "bfloat16", |
| "vocab_size": 257216 |
| }, |
| "timm_model_ids": [ |
| "vit_so400m_patch14_siglip_224" |
| ], |
| "timm_override_act_layers": [ |
| null |
| ], |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.44.2", |
| "use_fused_vision_backbone": false, |
| "vision_backbone_id": "siglip-vit-so400m" |
| } |
|
|
