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
|