nsorros commited on
Commit
1ca0f10
1 Parent(s): ccbf044
Files changed (1) hide show
  1. custom_model.py +1 -1
custom_model.py CHANGED
@@ -9,5 +9,5 @@ class Model(PreTrainedModel):
9
  self.model = AutoModel.from_pretrained("bert-base-uncased")
10
 
11
  def forward(self, **inputs):
12
- outs = self.model(inputs)
13
  return outs
 
9
  self.model = AutoModel.from_pretrained("bert-base-uncased")
10
 
11
  def forward(self, **inputs):
12
+ outs = self.model(**inputs)
13
  return outs