bourdoiscatie
commited on
Commit
•
d4f4d94
1
Parent(s):
037e2b8
Update modeling_flash_t5.py
Browse files- modeling_flash_t5.py +1 -2
modeling_flash_t5.py
CHANGED
@@ -514,7 +514,6 @@ class FlashT5PreTrainedModel(PreTrainedModel):
|
|
514 |
|
515 |
|
516 |
class FlashT5Model(FlashT5PreTrainedModel):
|
517 |
-
|
518 |
def __init__(self, config: FlashT5Config):
|
519 |
super().__init__(config)
|
520 |
self.shared = nn.Embedding(config.vocab_size, config.d_model)
|
@@ -837,4 +836,4 @@ class FlashT5EncoderModel(FlashT5PreTrainedModel):
|
|
837 |
return_dict=return_dict,
|
838 |
)
|
839 |
|
840 |
-
return encoder_outputs
|
|
|
514 |
|
515 |
|
516 |
class FlashT5Model(FlashT5PreTrainedModel):
|
|
|
517 |
def __init__(self, config: FlashT5Config):
|
518 |
super().__init__(config)
|
519 |
self.shared = nn.Embedding(config.vocab_size, config.d_model)
|
|
|
836 |
return_dict=return_dict,
|
837 |
)
|
838 |
|
839 |
+
return encoder_outputs
|