Deprecate models (#24787)
* Deprecate some models * Fix imports * Fix inits too * Remove tests * Add deprecated banner to documentation * Remove from init * Fix auto classes * Style * Remote upgrade strategy 1 * Remove site package cache * Revert this part * Fix typo... * Update utils * Update docs/source/en/model_doc/bort.md Co-authored-by: Lysandre Debut <lysandre.debut@reseau.eseo.fr> * Address review comments * With all files saved --------- Co-authored-by: Lysandre Debut <lysandre.debut@reseau.eseo.fr>
This commit is contained in:
@@ -400,6 +400,8 @@ def check_model_list():
|
||||
models_dir = os.path.join(PATH_TO_TRANSFORMERS, "models")
|
||||
_models = []
|
||||
for model in os.listdir(models_dir):
|
||||
if model == "deprecated":
|
||||
continue
|
||||
model_dir = os.path.join(models_dir, model)
|
||||
if os.path.isdir(model_dir) and "__init__.py" in os.listdir(model_dir):
|
||||
_models.append(model)
|
||||
@@ -445,6 +447,8 @@ def get_model_modules():
|
||||
]
|
||||
modules = []
|
||||
for model in dir(transformers.models):
|
||||
if model == "deprecated":
|
||||
continue
|
||||
# There are some magic dunder attributes in the dir, we ignore them
|
||||
if not model.startswith("__"):
|
||||
model_module = getattr(transformers.models, model)
|
||||
@@ -767,6 +771,8 @@ def check_objects_being_equally_in_main_init():
|
||||
obj = getattr(transformers, attr)
|
||||
if hasattr(obj, "__module__"):
|
||||
module_path = obj.__module__
|
||||
if "models.deprecated" in module_path:
|
||||
continue
|
||||
module_name = module_path.split(".")[-1]
|
||||
module_dir = ".".join(module_path.split(".")[:-1])
|
||||
if (
|
||||
|
||||
Reference in New Issue
Block a user