Update configuration_scgpt.py
Browse files- 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,
|