kunjcr2 commited on
Commit
0ad3934
·
verified ·
1 Parent(s): 36d5852

Update configuration_medassistgpt.py

Browse files
Files changed (1) hide show
  1. configuration_medassistgpt.py +4 -4
configuration_medassistgpt.py CHANGED
@@ -6,14 +6,14 @@ class MedAssistGPTConfig(PretrainedConfig):
6
  def __init__(
7
  self,
8
  vocab_size=50281,
9
- d_model=512,
10
  n_heads=16,
11
- gqa_groups=4,
12
  max_len=1024,
13
- d_ff=2048,
14
  eps=1e-5,
15
  dropout_p=0.1,
16
- blocks=16,
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)