msy127 commited on
Commit
e08793e
·
1 Parent(s): 41ecf42

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -71,7 +71,11 @@ llm = CustomInferenceClient(model_name=model_name, hf_token=hf_token, kwargs=kwa
71
 
72
  # 임베딩 객체 생성
73
  from langchain.embeddings import HuggingFaceInstructEmbeddings
74
- embeddings = HuggingFaceInstructEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2", model_kwargs={"device": "cuda"})
 
 
 
 
75
 
76
  # 벡터DB 로드
77
  path_work ='.'
 
71
 
72
  # 임베딩 객체 생성
73
  from langchain.embeddings import HuggingFaceInstructEmbeddings
74
+ embeddings = HuggingFaceInstructEmbeddings(
75
+ model_name="sentence-transformers/all-MiniLM-L6-v2",
76
+ cache_folder="./sentence-transformers/all-MiniLM-L6-v2",
77
+ model_kwargs={"device": "auto"}
78
+ )
79
 
80
  # 벡터DB 로드
81
  path_work ='.'