NousResearch
/

Yarn-Mistral-7b-128k

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

emozilla commited on Nov 1, 2023

Commit

c17e1a8

•

1 Parent(s): d0d06db

Update config.json

Files changed (1) hide show

config.json +5 -1

config.json CHANGED Viewed

@@ -1,8 +1,12 @@
 {
-  "_name_or_path": "/p/scratch/ccstdl/quesnelle1/output/yarn-mistral-7b-16k-scale-8-constant-lr-1e6-step1000-scale-16/step_1500",
   "architectures": [
     "MistralForCausalLM"
   ],
   "bos_token_id": 1,
   "eos_token_id": 2,
   "hidden_act": "silu",

 {
+  "_name_or_path": "NousResearch/Yarn-Mistral-7b-128k",
   "architectures": [
     "MistralForCausalLM"
   ],
+  "auto_map": {
+    "AutoConfig": "configuration_mistral.MistralConfig",
+    "AutoModelForCausalLM": "modeling_mistral_yarn.MistralForCausalLM"
+  },
   "bos_token_id": 1,
   "eos_token_id": 2,
   "hidden_act": "silu",