radames commited on
Commit
1d7d54e
1 Parent(s): 1ac1ce8

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -10,7 +10,7 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
10
  class PreTrainedPipeline():
11
  def __init__(self, path=""):
12
  # load the optimized model
13
- self.model_path = os.path.join("", '.')
14
  self.tokenizer = AutoTokenizer.from_pretrained(self.model_path)
15
  self.model = AutoModel.from_pretrained(self.model_path)
16
  self.model.eval()
 
10
  class PreTrainedPipeline():
11
  def __init__(self, path=""):
12
  # load the optimized model
13
+ self.model_path = os.path.join(path, '.')
14
  self.tokenizer = AutoTokenizer.from_pretrained(self.model_path)
15
  self.model = AutoModel.from_pretrained(self.model_path)
16
  self.model.eval()