| { | |
| "_name_or_path": "/data/moran/outputs/models/image-captioning-rlhf/karpathy_split/blip_base_coco_am_1_bertscore_1_bart_nli_reward_model_5_refs/step_349/", | |
| "architectures": [ | |
| "BlipForConditionalGeneration" | |
| ], | |
| "image_text_hidden_size": 256, | |
| "initializer_factor": 1.0, | |
| "initializer_range": 0.02, | |
| "logit_scale_init_value": 2.6592, | |
| "model_type": "blip", | |
| "projection_dim": 512, | |
| "text_config": { | |
| "initializer_factor": 1.0, | |
| "model_type": "blip_text_model", | |
| "num_attention_heads": 12 | |
| }, | |
| "torch_dtype": "float16", | |
| "transformers_version": "4.34.1", | |
| "vision_config": { | |
| "dropout": 0.0, | |
| "initializer_factor": 1.0, | |
| "initializer_range": 0.02, | |
| "model_type": "blip_vision_model", | |
| "num_channels": 3 | |
| } | |
| } | |