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 |
'''
|