From 3f707746fa39cf0841191201151fc42164d3f724 Mon Sep 17 00:00:00 2001
From: chflame163 <chflame@163.com>
Date: Tue, 3 Sep 2024 18:44:28 +0800
Subject: [PATCH] fix bug of SAM2 Ultra,SAM2 Video Ultra noses cache model

---
 py/sam_2_ultrl.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/py/sam_2_ultrl.py b/py/sam_2_ultrl.py
index a50173d..42a3c28 100644
--- a/py/sam_2_ultrl.py
+++ b/py/sam_2_ultrl.py
@@ -357,7 +357,10 @@ class LS_SAM2_ULTRA:
             try:
                 model.to(offload_device)
             except:
-                model.model.to(offload_device)
+                try:
+                    model.model.to(offload_device)
+                except:
+                    pass
         else:
             del model
             clear_memory()
@@ -657,8 +660,11 @@ class LS_SAM2_VIDEO_ULTRA:
                 v_model.to(offload_device)
                 s_model.to(offload_device)
             except:
-                v_model.model.to(offload_device)
-                s_model.model.to(offload_device)
+                try:
+                    v_model.model.to(offload_device)
+                    s_model.model.to(offload_device)
+                except:
+                    pass
         else:
             del v_model
             del s_model