haritzpuerto commited on
Commit
127b15f
1 Parent(s): 9f26017

Update inference.py

Browse files
Files changed (1) hide show
  1. 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) # change to 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
  '''