ayjays132 commited on
Commit
4323c95
1 Parent(s): c371975

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -1,7 +1,7 @@
1
  ---
2
  activation_function: gelu_new
3
  architectures:
4
- - GPT2LMHeadModel
5
  attn_pdrop: 0.1
6
  bos_token_id: 50256
7
  embd_pdrop: 0.1
 
1
  ---
2
  activation_function: gelu_new
3
  architectures:
4
+ - GPT2Model
5
  attn_pdrop: 0.1
6
  bos_token_id: 50256
7
  embd_pdrop: 0.1