| { | |
| "_attn_implementation_autoset": true, | |
| "architectures": [ | |
| "SafeLlavaLlamaForCausalLM" | |
| ], | |
| "attention_bias": false, | |
| "attention_dropout": 0.0, | |
| "bos_token_id": 1, | |
| "delay_load_vision_tower": true, | |
| "dtype": "bfloat16", | |
| "eos_token_id": 2, | |
| "freeze_mm_mlp_adapter": false, | |
| "freeze_mm_vision_resampler": false, | |
| "head_dim": 128, | |
| "hidden_act": "silu", | |
| "hidden_size": 4096, | |
| "image_aspect_ratio": "pad", | |
| "initializer_range": 0.02, | |
| "intermediate_size": 11008, | |
| "max_length": 4096, | |
| "max_position_embeddings": 4096, | |
| "mlp_bias": false, | |
| "mm_hidden_size": 1024, | |
| "mm_patch_merge_type": "flat", | |
| "mm_projector_lr": 2e-05, | |
| "mm_projector_type": "mlp2x_gelu", | |
| "mm_resampler_type": null, | |
| "mm_use_im_patch_token": false, | |
| "mm_use_im_start_end": false, | |
| "mm_vision_select_feature": "patch", | |
| "mm_vision_select_layer": -2, | |
| "mm_vision_tower": "openai/clip-vit-large-patch14-336", | |
| "model_type": "safe_llava_llama", | |
| "num_attention_heads": 32, | |
| "num_hidden_layers": 32, | |
| "num_key_value_heads": 32, | |
| "pad_token_id": 0, | |
| "pooling_method": "mean", | |
| "pretraining_tp": 1, | |
| "rms_norm_eps": 1e-05, | |
| "rope_scaling": null, | |
| "rope_theta": 10000.0, | |
| "safety_categories": [ | |
| "safe", | |
| "gender", | |
| "race", | |
| "religion", | |
| "harassment", | |
| "disability_discrimination", | |
| "drug_crime", | |
| "property_crime", | |
| "facial_data", | |
| "identity_data", | |
| "physical_self_injury", | |
| "suicide", | |
| "animal_abuse", | |
| "obscene_gestures", | |
| "physical_altercation", | |
| "terrorism", | |
| "weapon_related_violence", | |
| "sexual_content", | |
| "financial_advice", | |
| "medical_advice" | |
| ], | |
| "safety_head_hidden_scale": 0.5, | |
| "safety_num_hidden_layers": 1, | |
| "tie_word_embeddings": false, | |
| "tokenizer_model_max_length": 2048, | |
| "tokenizer_padding_side": "right", | |
| "transformers_version": "4.57.0", | |
| "tune_mm_mlp_adapter": false, | |
| "tune_mm_vision_resampler": false, | |
| "unfreeze_mm_vision_tower": true, | |
| "use_cache": true, | |
| "use_img_safety_meta_token": false, | |
| "use_mm_proj": true, | |
| "use_total_safety_meta_token": false, | |
| "use_txt_safety_meta_token": false, | |
| "vocab_size": 32000, | |
| "auto_map": { | |
| "AutoConfig": "modeling_safellava.SafetyConfig", | |
| "AutoModelForCausalLM": "modeling_safellava.SafeLlavaLlamaForCausalLM" | |
| } | |
| } |