Spaces:
Sleeping
Sleeping
Anigor66
commited on
Commit
·
548ad6f
1
Parent(s):
17a1674
Fixed errors
Browse files
app.py
CHANGED
|
@@ -57,8 +57,9 @@ torch.load = patched_torch_load
|
|
| 57 |
print(f"Loading SAM model ({MODEL_TYPE}) from downloaded checkpoint...")
|
| 58 |
|
| 59 |
try:
|
|
|
|
| 60 |
torch.load = patched_torch_load
|
| 61 |
-
sam = sam_model_registry[MODEL_TYPE](checkpoint=MODEL_CHECKPOINT)
|
| 62 |
finally:
|
| 63 |
torch.load = original_torch_load
|
| 64 |
|
|
|
|
| 57 |
print(f"Loading SAM model ({MODEL_TYPE}) from downloaded checkpoint...")
|
| 58 |
|
| 59 |
try:
|
| 60 |
+
# Ensure we always load onto CPU when no GPU is available
|
| 61 |
torch.load = patched_torch_load
|
| 62 |
+
sam = sam_model_registry[MODEL_TYPE](checkpoint=MODEL_CHECKPOINT)
|
| 63 |
finally:
|
| 64 |
torch.load = original_torch_load
|
| 65 |
|