Spaces:
Runtime error
Runtime error
Change in LangEfficientSAM
Browse files
lang_efficient_sam/LangEfficientSAM.py
CHANGED
@@ -30,7 +30,7 @@ class LangEfficientSAM:
|
|
30 |
def __init__(self, device=torch.device("cuda" if torch.cuda.is_available() else "cpu")):
|
31 |
self.device = device
|
32 |
print("Device:", self.device)
|
33 |
-
if self.device == "cpu":
|
34 |
self.sam_efficient = torch.jit.load('./models/efficientsam_s_cpu.jit')
|
35 |
else:
|
36 |
self.sam_efficient = torch.jit.load('./models/efficientsam_s_gpu.jit')
|
|
|
30 |
def __init__(self, device=torch.device("cuda" if torch.cuda.is_available() else "cpu")):
|
31 |
self.device = device
|
32 |
print("Device:", self.device)
|
33 |
+
if self.device == torch.device("cpu"):
|
34 |
self.sam_efficient = torch.jit.load('./models/efficientsam_s_cpu.jit')
|
35 |
else:
|
36 |
self.sam_efficient = torch.jit.load('./models/efficientsam_s_gpu.jit')
|