regisss
/

llama2-70b-fused-qkv-mlperf

Text Generation

Model card Files Files and versions Community

regisss HF staff commited on Feb 14

Commit

d5c7141

•

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"
   }
 }