From b71363a727bc35b3ac497df92d3111b9fe449b7a Mon Sep 17 00:00:00 2001
From: xiaoyuxi <xiaoyuxi@bytedance.com>
Date: Tue, 8 Jul 2025 18:11:58 +0800
Subject: [PATCH] PAPERMD

---
 models/SpaTrackV2/models/predictor.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/models/SpaTrackV2/models/predictor.py b/models/SpaTrackV2/models/predictor.py
index 10dd942..a6a2861 100644
--- a/models/SpaTrackV2/models/predictor.py
+++ b/models/SpaTrackV2/models/predictor.py
@@ -27,7 +27,8 @@ class Predictor(torch.nn.Module):
 
     def to(self, device: Union[str, torch.device]):
         self.spatrack.to(device)
-        self.spatrack.base_model.to(device)
+        if self.spatrack.base_model is not None:
+            self.spatrack.base_model.to(device)
 
     @classmethod
     def from_pretrained(