Spaces:
Starting
on
T4
Starting
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -128,7 +128,10 @@ def load_models_and_documents():
|
|
128 |
n_gpu_layers=32
|
129 |
)
|
130 |
|
131 |
-
dense_model = OptimumEncoder(
|
|
|
|
|
|
|
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',
|