Spaces:
Running
Running
change torch device to cpu only
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ import faiss
|
|
| 8 |
|
| 9 |
|
| 10 |
# Init similarity search AI model and processor
|
| 11 |
-
device = torch.device("
|
| 12 |
processor = AutoImageProcessor.from_pretrained("facebook/dinov2-large")
|
| 13 |
model = AutoModel.from_pretrained("facebook/dinov2-large")
|
| 14 |
model.config.return_dict = False # Set return_dict to False for JIT tracing
|
|
|
|
| 8 |
|
| 9 |
|
| 10 |
# Init similarity search AI model and processor
|
| 11 |
+
device = torch.device("cpu")
|
| 12 |
processor = AutoImageProcessor.from_pretrained("facebook/dinov2-large")
|
| 13 |
model = AutoModel.from_pretrained("facebook/dinov2-large")
|
| 14 |
model.config.return_dict = False # Set return_dict to False for JIT tracing
|