bourdoiscatie commited on
Commit
bbd070b
1 Parent(s): 2a4f976

Update modeling_flash_t5.py

Browse files
Files changed (1) hide show
  1. 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)