Spaces:
Running
Running
chore: Update device for HuggingFaceBgeEmbeddings to "cuda:0"
Browse files
app.py
CHANGED
@@ -104,7 +104,7 @@ print(f"총 도큐먼트 개수: {len(combined_documents)}")
|
|
104 |
store = LocalFileStore("~/.cache/embedding")
|
105 |
embeddings = HuggingFaceBgeEmbeddings(
|
106 |
model_name="BAAI/bge-m3",
|
107 |
-
model_kwargs={"device": "
|
108 |
encode_kwargs={"normalize_embeddings": True},
|
109 |
)
|
110 |
cached_embeddings = CacheBackedEmbeddings.from_bytes_store(
|
|
|
104 |
store = LocalFileStore("~/.cache/embedding")
|
105 |
embeddings = HuggingFaceBgeEmbeddings(
|
106 |
model_name="BAAI/bge-m3",
|
107 |
+
model_kwargs={"device": "cuda:0"},
|
108 |
encode_kwargs={"normalize_embeddings": True},
|
109 |
)
|
110 |
cached_embeddings = CacheBackedEmbeddings.from_bytes_store(
|