Spaces:
Starting
on
T4
Starting
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -134,7 +134,7 @@ def load_models_and_documents():
|
|
134 |
)
|
135 |
|
136 |
dense_model = OptimumEncoder(
|
137 |
-
device=
|
138 |
cache_dir=os.getenv('HF_HOME')
|
139 |
)
|
140 |
|
|
|
134 |
)
|
135 |
|
136 |
dense_model = OptimumEncoder(
|
137 |
+
device="cuda",
|
138 |
cache_dir=os.getenv('HF_HOME')
|
139 |
)
|
140 |
|