bourdoiscatie
commited on
Commit
•
bbd070b
1
Parent(s):
2a4f976
Update modeling_flash_t5.py
Browse files- modeling_flash_t5.py +1 -0
modeling_flash_t5.py
CHANGED
@@ -514,6 +514,7 @@ class FlashT5PreTrainedModel(PreTrainedModel):
|
|
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)
|
|
|
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)
|