Upload model
Browse files- config.json +1 -1
- model.safetensors +1 -1
- modeling_transformer.py +1 -2
config.json
CHANGED
|
@@ -4,7 +4,7 @@
|
|
| 4 |
],
|
| 5 |
"auto_map": {
|
| 6 |
"AutoConfig": "configuration_transformer.transformerConfig",
|
| 7 |
-
"
|
| 8 |
},
|
| 9 |
"batch_size": 64,
|
| 10 |
"dropout": 0.1,
|
|
|
|
| 4 |
],
|
| 5 |
"auto_map": {
|
| 6 |
"AutoConfig": "configuration_transformer.transformerConfig",
|
| 7 |
+
"AutoModel": "modeling_transformer.transformerModel"
|
| 8 |
},
|
| 9 |
"batch_size": 64,
|
| 10 |
"dropout": 0.1,
|
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 250204
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:638eca9017e20f9ab41086077abcf0c88eee7457a25b89ff5b4d5d4b958b7be9
|
| 3 |
size 250204
|
modeling_transformer.py
CHANGED
|
@@ -335,8 +335,7 @@ class TransformerDecoder(AttentionDecoder):
|
|
| 335 |
|
| 336 |
|
| 337 |
class transformerModel(PreTrainedModel):
|
| 338 |
-
|
| 339 |
-
Defined in :numref:`sec_encoder-decoder`"""
|
| 340 |
def __init__(self, config):
|
| 341 |
super().__init__(config)
|
| 342 |
self.encoder = TransformerEncoder(
|
|
|
|
| 335 |
|
| 336 |
|
| 337 |
class transformerModel(PreTrainedModel):
|
| 338 |
+
|
|
|
|
| 339 |
def __init__(self, config):
|
| 340 |
super().__init__(config)
|
| 341 |
self.encoder = TransformerEncoder(
|