| { |
| "_name_or_path": "alfiannajih/g-retriever", |
| "architectures": [ |
| "GRetrieverModel" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "auto_map": { |
| "AutoConfig": "alfiannajih/g-retriever--g_retriever_config.GRetrieverConfig", |
| "AutoModelForCausalLM": "alfiannajih/g-retriever--g_retriever_model.GRetrieverModel" |
| }, |
| "bos_id": [ |
| 128000, |
| 128006, |
| 882, |
| 128007 |
| ], |
| "bos_token_id": 128000, |
| "custom_pipelines": { |
| "g-retriever-task": { |
| "default": { |
| "model": { |
| "pt": [ |
| "alfiannajih/g-retriever", |
| "dee4a24" |
| ] |
| } |
| }, |
| "impl": "g_retriever_pipeline.GRetrieverPipeline", |
| "pt": [ |
| "AutoModelForCausalLM" |
| ], |
| "tf": [] |
| } |
| }, |
| "eos_token_id": 128040, |
| "gnn_dropout": 0, |
| "gnn_hidden_dim": 1024, |
| "gnn_in_dim": 768, |
| "gnn_num_heads": 4, |
| "gnn_num_layers": 4, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "initializer_range": 0.02, |
| "intermediate_size": 14336, |
| "max_new_tokens": 256, |
| "max_position_embeddings": 131072, |
| "max_txt_len": 1024, |
| "mlp_bias": false, |
| "model_type": "llama", |
| "num_attention_heads": 32, |
| "num_hidden_layers": 32, |
| "num_key_value_heads": 8, |
| "pad_token_id": 128040, |
| "pretraining_tp": 1, |
| "rms_norm_eps": 1e-05, |
| "rope_scaling": { |
| "factor": 8.0, |
| "high_freq_factor": 4.0, |
| "low_freq_factor": 1.0, |
| "original_max_position_embeddings": 8192, |
| "rope_type": "llama3" |
| }, |
| "rope_theta": 500000.0, |
| "tie_word_embeddings": false, |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.44.2", |
| "use_cache": true, |
| "vocab_size": 128256 |
| } |
