diff --git a/dependency_checker.py b/dependency_checker.py index 25966f6..f465963 100644 --- a/dependency_checker.py +++ b/dependency_checker.py @@ -46,7 +46,7 @@ def handle_model_info(ckpt_path): item = { "filename": filename, - "save_path": save_path, + "save_path": os.path.relpath(save_path, BASE_PATH), "urls": urls, } return model_id, item