regisss
/

llama2-70b-fused-qkv-mlperf

Text Generation

Model card Files Files and versions Community

regisss HF staff commited on Feb 14, 2024

Commit

f939c73

•

1 Parent(s): ddcfa9e

Add auto_map

Files changed (1) hide show

config.json +5 -1

config.json CHANGED Viewed

@@ -24,5 +24,9 @@
   "torch_dtype": "bfloat16",
   "transformers_version": "4.37.2",
   "use_cache": false,
-  "vocab_size": 32000
 }

   "torch_dtype": "bfloat16",
   "transformers_version": "4.37.2",
   "use_cache": false,
+  "vocab_size": 32000,
+  "auto_map": {
+	"AutoConfig": "regisss/llama2-70b-fused-qkv-mlperf--CustomLlamaConfig",
+	"AutoModelForCausalLM": "regisss/llama2-70b-fused-qkv-mlperf--CustomLlamaForCausalLM",
+  },
 }