uform-gen-chat / config.json
kimihailv's picture
Update config.json
9887a97
raw
history blame contribute delete
742 Bytes
{
"_name_or_path": "uform-gen-chat",
"architectures": [
"VLMForCausalLM"
],
"center_crop": true,
"image_encoder_embedding_dim": 256,
"image_encoder_hidden_size": 768,
"image_encoder_num_heads": 12,
"image_encoder_num_layers": 12,
"image_encoder_patch_size": 16,
"image_encoder_pooling": "cls",
"image_pooler_intermediate_size": 5504,
"image_pooler_num_attn_heads": 16,
"image_pooler_num_latents": 196,
"image_size": 224,
"image_token_id": 32002,
"initializer_range": 0.02,
"model_type": "vlm",
"text_decoder_name_or_path": "princeton-nlp/Sheared-LLaMA-1.3B",
"tokenizer_name_or_path": "unum-cloud/uform-gen-chat",
"torch_dtype": "float32",
"transformers_version": "4.36.2",
"use_cache": true
}