Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
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": "
|
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 로드
|