From 44ac4cd0f89aa591477cbdb07d55ca40e974b3d1 Mon Sep 17 00:00:00 2001
From: wl-zhao <zhaowl20@mails.tsinghua.edu.cn>
Date: Sun, 20 Oct 2024 10:19:22 +0800
Subject: [PATCH] update relpath of models

---
 dependency_checker.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/dependency_checker.py b/dependency_checker.py
index f465963..fde921a 100644
--- a/dependency_checker.py
+++ b/dependency_checker.py
@@ -23,7 +23,7 @@ def handle_model_info(ckpt_path):
     ckpt_path = windows_to_linux_path(ckpt_path)
     filename = os.path.basename(ckpt_path)
     dirname = os.path.dirname(ckpt_path)
-    save_path = dirname.split('/', 1)[1]
+    save_path = os.path.dirname(os.path.relpath(ckpt_path, MODELS_DIR))
     metadata_path = ckpt_path + ".json"
     if os.path.isfile(metadata_path):
         metadata = json.load(open(metadata_path))
@@ -46,7 +46,7 @@ def handle_model_info(ckpt_path):
         
     item = {
         "filename": filename,
-        "save_path": os.path.relpath(save_path, BASE_PATH),
+        "save_path": save_path,
         "urls": urls,
     }
     return model_id, item