jwieting commited on
Commit
52f8006
1 Parent(s): 4ffede5

Update modeling_sim.py

Browse files
Files changed (1) hide show
  1. modeling_sim.py +1 -1
modeling_sim.py CHANGED
@@ -12,4 +12,4 @@ class SimModel(PreTrainedModel):
12
 
13
  def forward(self, **input_args):
14
  print(**input_args)
15
- return self.t5_encoder(**input_args).last_hidden_state
 
12
 
13
  def forward(self, **input_args):
14
  print(**input_args)
15
+ return self.encoder(**input_args).last_hidden_state