Nick Sorros commited on
Commit ·
1ca0f10
1
Parent(s): ccbf044
add model
Browse files- 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
|