regisss
/

llama2-70b-fused-qkv-mlperf

Text Generation

Model card Files Files and versions

regisss HF Staff commited on Feb 14, 2024

Commit

d5c7141

·

verified ·

1 Parent(s): 54ce86e

Fix

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -26,7 +26,7 @@
   "use_cache": false,
   "vocab_size": 32000,
   "auto_map": {
-	"AutoConfig": "regisss/llama2-70b-fused-qkv-mlperf--CustomLlamaConfig",
-	"AutoModelForCausalLM": "regisss/llama2-70b-fused-qkv-mlperf--CustomLlamaForCausalLM"
   }
 }

   "use_cache": false,
   "vocab_size": 32000,
   "auto_map": {
+	"AutoConfig": "modeling_llama.CustomLlamaConfig",
+	"AutoModelForCausalLM": "modeling_llama.CustomLlamaForCausalLM"
   }
 }