Change in LangEfficientSAM

lang_efficient_sam/LangEfficientSAM.py

@@ -30,7 +30,7 @@ class LangEfficientSAM:
     def __init__(self, device=torch.device("cuda" if torch.cuda.is_available() else "cpu")):
         self.device = device
         print("Device:", self.device)
-        if self.device == "cpu":
+        if self.device == torch.device("cpu"):
             self.sam_efficient = torch.jit.load('./models/efficientsam_s_cpu.jit')
         else:
             self.sam_efficient = torch.jit.load('./models/efficientsam_s_gpu.jit')