memorease commited on
Commit
6b2bb5d
·
verified ·
1 Parent(s): 5cbac8d

Update model_index.json

Browse files
Files changed (1) hide show
  1. model_index.json +4 -1
model_index.json CHANGED
@@ -1,3 +1,6 @@
1
  {
2
- "auto_model": "AutoModelForCausalLM"
 
 
 
3
  }
 
1
  {
2
+ "auto_map": {
3
+ "AutoConfig": "llama.LlamaConfig",
4
+ "AutoModelForCausalLM": "llama.LlamaForCausalLM"
5
+ }
6
  }