bourdoiscatie commited on
Commit
d4f4d94
1 Parent(s): 037e2b8

Update modeling_flash_t5.py

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