muryshev commited on
Commit
aec62e6
1 Parent(s): 8e14974

removed token

Browse files
Files changed (1) hide show
  1. semantic_search.py +2 -2
semantic_search.py CHANGED
@@ -82,8 +82,8 @@ class SemanticSearch:
82
 
83
  def load_model(self):
84
  if hf_token and hf_model_name:
85
- self.tokenizer = AutoTokenizer.from_pretrained(hf_model_name, use_auth_token=True, token=hf_token)
86
- self.model = AutoModel.from_pretrained(hf_model_name, use_auth_token=True, token=hf_token).to(self.device)
87
  else:
88
  self.tokenizer = AutoTokenizer.from_pretrained(global_model_path)
89
  self.model = AutoModel.from_pretrained(global_model_path).to(self.device)
 
82
 
83
  def load_model(self):
84
  if hf_token and hf_model_name:
85
+ self.tokenizer = AutoTokenizer.from_pretrained(hf_model_name, use_auth_token=True)
86
+ self.model = AutoModel.from_pretrained(hf_model_name, use_auth_token=True).to(self.device)
87
  else:
88
  self.tokenizer = AutoTokenizer.from_pretrained(global_model_path)
89
  self.model = AutoModel.from_pretrained(global_model_path).to(self.device)