devve1 commited on
Commit
33834b7
1 Parent(s): 1bbf947

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -128,7 +128,10 @@ def load_models_and_documents():
128
  n_gpu_layers=32
129
  )
130
 
131
- dense_model = OptimumEncoder(cache_dir=os.getenv('HF_HOME'))
 
 
 
132
 
133
  sparse_model = Bm42(
134
  'Qdrant/all_miniLM_L6_v2_with_attentions',
 
128
  n_gpu_layers=32
129
  )
130
 
131
+ dense_model = OptimumEncoder(
132
+ device='cuda',
133
+ cache_dir=os.getenv('HF_HOME')
134
+ )
135
 
136
  sparse_model = Bm42(
137
  'Qdrant/all_miniLM_L6_v2_with_attentions',