lchakkei commited on
Commit
7d0f09a
·
verified ·
1 Parent(s): fb06509

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -72,7 +72,7 @@ class EndpointHandler():
72
 
73
  # Create Text-Embedding Model
74
  embedding_function = HuggingFaceBgeEmbeddings(
75
- model_name="DMetaSoul/Dmeta-embedding",
76
  model_kwargs={'device': 'cuda'},
77
  encode_kwargs={'normalize_embeddings': True}
78
  )
 
72
 
73
  # Create Text-Embedding Model
74
  embedding_function = HuggingFaceBgeEmbeddings(
75
+ model_name="mixedbread-ai/mxbai-embed-large-v1",
76
  model_kwargs={'device': 'cuda'},
77
  encode_kwargs={'normalize_embeddings': True}
78
  )