anpigon commited on
Commit
6cf0662
1 Parent(s): ed4d993

chore: Update device for HuggingFaceBgeEmbeddings to "cuda:0"

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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": "mps"},
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(