From 19ec398bcf0d44666e86bcb28e3c5b7100848469 Mon Sep 17 00:00:00 2001
From: xiaoyuxi <xiaoyuxi@bytedance.com>
Date: Tue, 8 Jul 2025 16:37:13 +0800
Subject: [PATCH] first-commit

---
 inference.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inference.py b/inference.py
index 906d9ea..ebe0747 100644
--- a/inference.py
+++ b/inference.py
@@ -46,7 +46,7 @@ if __name__ == "__main__":
     VGGT_DIR = os.environ["VGGT_DIR"]
     assert os.path.exists(VGGT_DIR), f"VGGT_DIR {VGGT_DIR} does not exist"
     front_track = VGGT4Track()
-    front_track.load_state_dict(torch.load(VGGT_DIR)["model"], strict=False)
+    front_track.load_state_dict(torch.load(VGGT_DIR), strict=False)
     front_track.eval()
     front_track = front_track.to("cuda")