haritzpuerto
commited on
Commit
•
127b15f
1
Parent(s):
9f26017
Update inference.py
Browse files- inference.py +1 -1
inference.py
CHANGED
@@ -14,7 +14,7 @@ class PredictionRequest():
|
|
14 |
class MetaQA():
|
15 |
def __init__(self, path_to_model):
|
16 |
self.metaqa_model = MetaQA_Model.from_pretrained(path_to_model)
|
17 |
-
self.tokenizer = AutoTokenizer.from_pretrained(path_to_model)
|
18 |
|
19 |
def run_metaqa(self, request: PredictionRequest):
|
20 |
'''
|
|
|
14 |
class MetaQA():
|
15 |
def __init__(self, path_to_model):
|
16 |
self.metaqa_model = MetaQA_Model.from_pretrained(path_to_model)
|
17 |
+
self.tokenizer = AutoTokenizer.from_pretrained(path_to_model)
|
18 |
|
19 |
def run_metaqa(self, request: PredictionRequest):
|
20 |
'''
|