Tonic commited on
Commit
ac88608
β€’
1 Parent(s): 668eaf5

fix sentence transformer latest version

Browse files
Files changed (1) hide show
  1. langchainapp.py +2 -2
langchainapp.py CHANGED
@@ -45,8 +45,8 @@ hf_token, yi_token = load_env_variables()
45
 
46
  tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token, trust_remote_code=True)
47
  nvidiamodel = AutoModel.from_pretrained(model_name, token=hf_token, trust_remote_code=True).to(device)
48
- nvidiamodel.set_pooling_include_prompt(include_prompt=False)
49
-
50
  def clear_cuda_cache():
51
  torch.cuda.empty_cache()
52
 
 
45
 
46
  tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token, trust_remote_code=True)
47
  nvidiamodel = AutoModel.from_pretrained(model_name, token=hf_token, trust_remote_code=True).to(device)
48
+ # nvidiamodel.set_pooling_include_prompt(include_prompt=False)
49
+
50
  def clear_cuda_cache():
51
  torch.cuda.empty_cache()
52