klemenk commited on
Commit
57b6477
·
verified ·
1 Parent(s): 18e4897

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -3,8 +3,8 @@
3
  "architectures": ["GPT"],
4
  "model_type": "ConnectomePruning.GPT",
5
  "auto_map": {
6
- "AutoConfig": "configuration_gpt.GPTConfig",
7
- "AutoModelForCausalLM": "modeling_gpt.GPT"
8
  },
9
  "vocab_size": 50304,
10
  "n_embd": 768,
 
3
  "architectures": ["GPT"],
4
  "model_type": "ConnectomePruning.GPT",
5
  "auto_map": {
6
+ "AutoConfig": "config.GPTConfig",
7
+ "AutoModelForCausalLM": "model.GPT"
8
  },
9
  "vocab_size": 50304,
10
  "n_embd": 768,