iitolstykh's picture
Update config.json
ff6d907 verified
{
"architectures": [
"MistralForCausalLM",
"MistralAIDetectorForSequenceClassification"
],
"auto_map": {
"AutoConfig": "configuration_gigacheck.GigaCheckConfig",
"AutoModel": "modeling_gigacheck.GigaCheckForDetection"
},
"conf_interval_thresh": 0.8,
"attention_dropout": 0.0,
"bos_token_id": 1,
"classifier_dropout": 0.1,
"detr_config": {
"aux_loss": true,
"center_noise_scale": 0.6,
"dab_detr": true,
"dec_layers": 3,
"dn_detr": true,
"enc_layers": 3,
"eos_coef": 0.1,
"extractor_dtype": "bfloat16",
"giou_loss_coef": 1.0,
"label_loss_coef": 2.0,
"losses": [],
"model_dim": 256,
"n_input_proj": 2,
"nheads": 8,
"noise_offset": 0.05,
"num_groups": 5,
"num_input_proj_layers": 2,
"num_queries": 45,
"position_embedding": "sine",
"proj_dropout": 0.5,
"query_initialization_method": "default",
"set_cost_class": 2,
"set_cost_giou": 1,
"set_cost_iou": 2,
"set_cost_reference": 1,
"set_cost_span": 5,
"span_loss_coef": 10.0,
"span_loss_type": "l1",
"special_ref_points": true,
"temperature": 10000,
"tgt_label_loss_coef": 0.0,
"tgt_loss_giou_coef": 3.0,
"tgt_loss_span_coef": 9.0,
"use_focal_loss": true,
"width_noise_scale": 0.7
},
"eos_token_id": 2,
"head_dim": null,
"hidden_act": "silu",
"hidden_size": 4096,
"id2label": {
"0": "ai",
"1": "human",
"2": "mixed"
},
"initializer_range": 0.02,
"intermediate_size": 14336,
"label2id": {
"LABEL_0": 0,
"LABEL_1": 1,
"LABEL_2": 2
},
"max_length": 1024,
"max_position_embeddings": 32768,
"model_type": "mistral",
"num_attention_heads": 32,
"num_hidden_layers": 32,
"num_key_value_heads": 8,
"pad_token_id": 0,
"pretrained_model_name_or_path": "mistralai/Mistral-7B-v0.3",
"rms_norm_eps": 1e-05,
"rope_theta": 1000000.0,
"sliding_window": null,
"tie_word_embeddings": false,
"torch_dtype": "float32",
"trained_classification_head": false,
"transformers_version": "4.55.0",
"use_cache": true,
"vocab_size": 32768,
"with_detr": true
}