TaylorAI
/

Flash-Llama-7B

Text Generation

text-generation-inference

Model card Files Files and versions

andersonbcdefg commited on Aug 26, 2023

Commit

a9ab032

·

1 Parent(s): 5784f0b

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -4,7 +4,7 @@
     "LlamaForCausalLM"
   ],
   "auto_map": {
-    "AutoModelForCausalLM": "togethercomputer/LLaMA-2-7B-32K--modeling_flash_llama.LlamaForCausalLM"
   },
   "bos_token_id": 1,
   "eos_token_id": 2,

     "LlamaForCausalLM"
   ],
   "auto_map": {
+    "AutoModelForCausalLM": "modeling_flash_llama.LlamaForCausalLM"
   },
   "bos_token_id": 1,
   "eos_token_id": 2,