Gleb Vinarskis
commited on
Commit
·
bde2f2a
1
Parent(s):
c0cd66f
debug
Browse files- modeling_stacked.py +1 -0
modeling_stacked.py
CHANGED
|
@@ -23,6 +23,7 @@ class ExtendedMultitaskModelForTokenClassification(PreTrainedModel):
|
|
| 23 |
print("11111111111111111111")
|
| 24 |
self.model = floret.load_model(self.config.filename)
|
| 25 |
print("22222222222222222222")
|
|
|
|
| 26 |
|
| 27 |
def predict(self, text, k=1):
|
| 28 |
predictions = self.model.predict(text, k)
|
|
|
|
| 23 |
print("11111111111111111111")
|
| 24 |
self.model = floret.load_model(self.config.filename)
|
| 25 |
print("22222222222222222222")
|
| 26 |
+
self.post_init()
|
| 27 |
|
| 28 |
def predict(self, text, k=1):
|
| 29 |
predictions = self.model.predict(text, k)
|