Upload GPTRefactForCausalLM (#7)
Browse files- Upload GPTRefactForCausalLM (336c052f7b9e0d991dbab8888b2cd1c6bc4d3759)
- configuration_gpt_refact.py +11 -12
- pytorch_model.bin +1 -1
configuration_gpt_refact.py
CHANGED
@@ -17,25 +17,24 @@ class GPTRefactConfig(PretrainedConfig):
|
|
17 |
|
18 |
def __init__(
|
19 |
self,
|
20 |
-
vocab_size=49216,
|
21 |
-
n_positions=
|
22 |
-
n_embd=
|
23 |
-
n_layer=
|
24 |
-
n_head=
|
25 |
-
|
26 |
-
|
27 |
-
embd_pdrop=0.1,
|
28 |
-
attn_pdrop=0.1,
|
29 |
layer_norm_epsilon=1e-5,
|
30 |
initializer_range=0.02,
|
31 |
scale_attn_weights=True,
|
32 |
use_cache=True,
|
33 |
bos_token_id=-1,
|
34 |
eos_token_id=0,
|
35 |
-
max_position_embeddings: int = 4096,
|
36 |
-
multi_query: bool = True,
|
37 |
attention_softmax_in_fp32=False,
|
38 |
scale_attention_softmax_in_fp32=False,
|
|
|
|
|
|
|
39 |
**kwargs,
|
40 |
):
|
41 |
self.vocab_size = vocab_size
|
@@ -43,7 +42,7 @@ class GPTRefactConfig(PretrainedConfig):
|
|
43 |
self.n_embd = n_embd
|
44 |
self.n_layer = n_layer
|
45 |
self.n_head = n_head
|
46 |
-
self.n_inner =
|
47 |
self.resid_pdrop = resid_pdrop
|
48 |
self.embd_pdrop = embd_pdrop
|
49 |
self.attn_pdrop = attn_pdrop
|
|
|
17 |
|
18 |
def __init__(
|
19 |
self,
|
20 |
+
vocab_size: int = 49216,
|
21 |
+
n_positions: int = 4096,
|
22 |
+
n_embd: int = 1024,
|
23 |
+
n_layer: int = 32,
|
24 |
+
n_head: int = 64,
|
25 |
+
max_position_embeddings: int = 4096,
|
26 |
+
multi_query: bool = True,
|
|
|
|
|
27 |
layer_norm_epsilon=1e-5,
|
28 |
initializer_range=0.02,
|
29 |
scale_attn_weights=True,
|
30 |
use_cache=True,
|
31 |
bos_token_id=-1,
|
32 |
eos_token_id=0,
|
|
|
|
|
33 |
attention_softmax_in_fp32=False,
|
34 |
scale_attention_softmax_in_fp32=False,
|
35 |
+
resid_pdrop=0.1,
|
36 |
+
embd_pdrop=0.1,
|
37 |
+
attn_pdrop=0.1,
|
38 |
**kwargs,
|
39 |
):
|
40 |
self.vocab_size = vocab_size
|
|
|
42 |
self.n_embd = n_embd
|
43 |
self.n_layer = n_layer
|
44 |
self.n_head = n_head
|
45 |
+
self.n_inner = None
|
46 |
self.resid_pdrop = resid_pdrop
|
47 |
self.embd_pdrop = embd_pdrop
|
48 |
self.attn_pdrop = attn_pdrop
|
pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 6343461637
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:eb422076b1a52027d21772a7c4cbb7365078c2dd489384a97d6ce61c8b7b7204
|
3 |
size 6343461637
|