Spaces:
Build error
Build error
fix sentence transformer latest version
Browse files- 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 |
|