From 1d015386aa08eac2fd2b989bcc06ad6994118bcb Mon Sep 17 00:00:00 2001
From: AI Lab <129358391+1038lab@users.noreply.github.com>
Date: Mon, 15 Dec 2025 23:00:05 -0800
Subject: [PATCH] Add files via upload

---
 sam2/utils/transforms.py | 20 +++++++++++++++-----
 1 file changed, 15 insertions(+), 5 deletions(-)

diff --git a/sam2/utils/transforms.py b/sam2/utils/transforms.py
index 3a8ceaf..6482839 100644
--- a/sam2/utils/transforms.py
+++ b/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)