Update configuration_medassistgpt.py
Browse files
configuration_medassistgpt.py
CHANGED
|
@@ -6,14 +6,14 @@ class MedAssistGPTConfig(PretrainedConfig):
|
|
| 6 |
def __init__(
|
| 7 |
self,
|
| 8 |
vocab_size=50281,
|
| 9 |
-
d_model=
|
| 10 |
n_heads=16,
|
| 11 |
-
gqa_groups=
|
| 12 |
max_len=1024,
|
| 13 |
-
d_ff=
|
| 14 |
eps=1e-5,
|
| 15 |
dropout_p=0.1,
|
| 16 |
-
blocks=
|
| 17 |
**kwargs,
|
| 18 |
):
|
| 19 |
super().__init__(**kwargs)
|
|
|
|
| 6 |
def __init__(
|
| 7 |
self,
|
| 8 |
vocab_size=50281,
|
| 9 |
+
d_model=1024,
|
| 10 |
n_heads=16,
|
| 11 |
+
gqa_groups=8,
|
| 12 |
max_len=1024,
|
| 13 |
+
d_ff=4096,
|
| 14 |
eps=1e-5,
|
| 15 |
dropout_p=0.1,
|
| 16 |
+
blocks=24,
|
| 17 |
**kwargs,
|
| 18 |
):
|
| 19 |
super().__init__(**kwargs)
|