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

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -72,9 +72,9 @@ llm = CustomInferenceClient(model_name=model_name, hf_token=hf_token, kwargs=kwa
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 로드
 
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": "cpu"}
78
  )
79
 
80
  # 벡터DB 로드