devve1 commited on
Commit
e93dcb1
1 Parent(s): 8cadb5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -219,8 +219,8 @@ def load_models_and_documents():
219
  [
220
  EngineArgs(
221
  model_name_or_path='mixedbread-ai/mxbai-embed-large-v1',
222
- engine='torch',
223
- device='cuda',
224
  embedding_dtype='float32',
225
  dtype='float16',
226
  pooling_method='cls',
 
219
  [
220
  EngineArgs(
221
  model_name_or_path='mixedbread-ai/mxbai-embed-large-v1',
222
+ engine='optimum',
223
+ device='cpu',
224
  embedding_dtype='float32',
225
  dtype='float16',
226
  pooling_method='cls',