diff --git a/dependency_checker.py b/dependency_checker.py index 05039b2..0f7d5fd 100644 --- a/dependency_checker.py +++ b/dependency_checker.py @@ -25,6 +25,8 @@ def handle_model_info(ckpt_path): model_id = metadata["id"] else: logging.info(f"computing sha256 of {ckpt_path}") + if not os.path.isfile(ckpt_path): + raise NotImplementedError(f"please install {ckpt_path} first!") model_id = compute_sha256(ckpt_path) data = { "id": model_id,