diff --git a/R/utils.R b/R/utils.R index 28cca311..c6d68002 100644 --- a/R/utils.R +++ b/R/utils.R @@ -12,7 +12,7 @@ download_and_cache <- function(url, redownload = FALSE, prefix = NULL) { if (!file.exists(path) || redownload) { # we should first download to a temporary file because - # download probalems could cause hard to debug errors. + # download problems could cause hard to debug errors. tmp <- tempfile(fileext = fs::path_ext(path)) on.exit({try({fs::file_delete(tmp)}, silent = TRUE)}, add = TRUE) diff --git a/tools/convert-models.py b/tools/convert-models.py index b70b3564..db61b692 100644 --- a/tools/convert-models.py +++ b/tools/convert-models.py @@ -146,6 +146,8 @@ def blob_exist(bucket_name, blob_name): 'yolo_v12_n': 'https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo12n.pt', 'yolo_v12_s': 'https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo12s.pt', 'yolo_v12_x': 'https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo12x.pt', + 'tabicl_classifier_v2': 'https://huggingface.co/jingang/TabICL/resolve/main/tabicl-classifier-v2-20260212.ckpt', + 'tabicl_regressor_v2': 'https://huggingface.co/jingang/TabICL/resolve/main/tabicl-regressor-v2-20260212.ckpt', } os.makedirs("models", exist_ok=True) @@ -172,8 +174,8 @@ def blob_exist(bucket_name, blob_name): if name.startswith("convnext_") and name.endswith("k"): m = m["model"] - # openmmlab models weights are embedded in a named object - if name.startswith("convnext_") and "_upernet_" in name: + # openmmlab and jingang models weights are embedded in a named object + if (name.startswith("convnext_") and "_upernet_" in name) or name.startswith("tabicl_"): m = m["state_dict"] for nm, par in m.items():