diff --git a/cccv/auto/config.py b/cccv/auto/config.py index c7e055b..4e6b602 100644 --- a/cccv/auto/config.py +++ b/cccv/auto/config.py @@ -1,6 +1,5 @@ import importlib.util import json -import warnings from pathlib import Path from typing import Any, Optional, Union @@ -24,14 +23,6 @@ def from_pretrained( :param model_dir: The path to cache the downloaded model configuration. Should be a full path. If None, use default cache path. :return: """ - if "pretrained_model_name" in kwargs: - warnings.warn( - "[CCCV] 'pretrained_model_name' is deprecated, please use 'pretrained_model_name_or_path' instead.", - DeprecationWarning, - stacklevel=2, - ) - pretrained_model_name_or_path = kwargs.pop("pretrained_model_name") - # 1. check if it's a registered config name, early return if found if isinstance(pretrained_model_name_or_path, ConfigType): pretrained_model_name_or_path = pretrained_model_name_or_path.value diff --git a/cccv/auto/model.py b/cccv/auto/model.py index 6fd5f4e..472d0da 100644 --- a/cccv/auto/model.py +++ b/cccv/auto/model.py @@ -86,7 +86,6 @@ def from_config( :param gh_proxy: The proxy for downloading from github release. Example: https://github.abskoop.workers.dev/ :return: """ - model = MODEL_REGISTRY.get(config.model) model = model( config=config,