Tonic commited on
Commit
7d4d809
·
1 Parent(s): 230772a

added langchain app for simplicity

Browse files
Files changed (1) hide show
  1. langchainapp.py +1 -1
langchainapp.py CHANGED
@@ -61,7 +61,7 @@ class MyEmbeddingFunction(EmbeddingFunction):
61
  self.intention_client = intention_client
62
  self.hf_embeddings = HuggingFaceInstructEmbeddings(
63
  model_name=model_name,
64
- model_kwargs={'device': 'cuda' if torch.cuda.is_available() else 'cpu'},
65
  encode_kwargs={'normalize_embeddings': True}
66
  )
67
 
 
61
  self.intention_client = intention_client
62
  self.hf_embeddings = HuggingFaceInstructEmbeddings(
63
  model_name=model_name,
64
+ model_kwargs={'device': 'cuda' if torch.cuda.is_available() else 'cpu', "trust_remote_code": True},
65
  encode_kwargs={'normalize_embeddings': True}
66
  )
67