devve1 commited on
Commit
5e45afc
1 Parent(s): 093ecda

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -137,15 +137,15 @@ def load_models_and_documents():
137
  type_v=GGML_TYPE_I8
138
  )
139
 
140
- dense_model = OptimumEncoder(
141
- device="cuda",
142
- cache_dir=os.getenv('HF_HOME')
143
- )
144
 
145
  sparse_model = SparseTextEmbedding(
146
  'Qdrant/bm42-all-minilm-l6-v2-attentions',
147
  cache_dir=os.getenv('HF_HOME'),
148
- providers=['CUDAExecutionProvider']
149
  )
150
 
151
  data_dir = os.path.join(os.getenv('HF_HOME'), 'nltk_data')
 
137
  type_v=GGML_TYPE_I8
138
  )
139
 
140
+ #dense_model = OptimumEncoder(
141
+ #device="cuda",
142
+ #cache_dir=os.getenv('HF_HOME')
143
+ #)
144
 
145
  sparse_model = SparseTextEmbedding(
146
  'Qdrant/bm42-all-minilm-l6-v2-attentions',
147
  cache_dir=os.getenv('HF_HOME'),
148
+ providers=['CPUExecutionProvider']
149
  )
150
 
151
  data_dir = os.path.join(os.getenv('HF_HOME'), 'nltk_data')