Update modeling_sim.py
Browse files- 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.
|
|
|
12 |
|
13 |
def forward(self, **input_args):
|
14 |
print(**input_args)
|
15 |
+
return self.encoder(**input_args).last_hidden_state
|