agemagician commited on
Commit
4be0483
·
verified ·
1 Parent(s): 499cd09

Update configuration_scgpt.py

Browse files
Files changed (1) hide show
  1. configuration_scgpt.py +5 -5
configuration_scgpt.py CHANGED
@@ -7,11 +7,11 @@ class ScgptConfig(PretrainedConfig):
7
 
8
  def __init__(
9
  self,
10
- ntoken: int,
11
- d_model: int,
12
- nhead: int,
13
- d_hid: int,
14
- nlayers: int,
15
  nlayers_cls: int = 3,
16
  n_cls: int = 1,
17
  vocab: Any = None,
 
7
 
8
  def __init__(
9
  self,
10
+ ntoken: int = 60697,
11
+ d_model: int = 512,
12
+ nhead: int = 8,
13
+ d_hid: int = 512,
14
+ nlayers: int = 12,
15
  nlayers_cls: int = 3,
16
  n_cls: int = 1,
17
  vocab: Any = None,