Wonder-Griffin commited on
Commit
694bdc2
1 Parent(s): 6cfc3be

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -22,7 +22,7 @@
22
  "max_computation_steps": 10,
23
  "max_len": 512,
24
  "memory_size": 512,
25
- "model_type": "gpt2",
26
  "n_embd": 512,
27
  "n_head": 8,
28
  "n_layer": 8,
 
22
  "max_computation_steps": 10,
23
  "max_len": 512,
24
  "memory_size": 512,
25
+ "model_type": "GPT2LMHeadModel",
26
  "n_embd": 512,
27
  "n_head": 8,
28
  "n_layer": 8,