Add files via upload

sam2/utils/transforms.py

@@ -27,13 +27,23 @@ class SAM2Transforms(nn.Module):
         self.mean = [0.485, 0.456, 0.406]
         self.std = [0.229, 0.224, 0.225]
         self.to_tensor = ToTensor()
-        self.transforms = torch.jit.script(
-            nn.Sequential(
-                Resize((self.resolution, self.resolution)),
-                Normalize(self.mean, self.std),
-            )
+        seq = nn.Sequential(
+            Resize((self.resolution, self.resolution)),
+            Normalize(self.mean, self.std),
         )
 
+        try:
+            self.transforms = torch.jit.script(seq)
+        except Exception:
+            self.transforms = seq
+
+        # self.transforms = torch.jit.script(
+        #     nn.Sequential(
+        #         Resize((self.resolution, self.resolution)),
+        #         Normalize(self.mean, self.std),
+        #     )
+        # )
+
     def __call__(self, x):
         x = self.to_tensor(x)
         return self.transforms(x)