From 546e7679e7f692ebeefcfc5063cec271a55bae20 Mon Sep 17 00:00:00 2001 From: Maria Khalusova Date: Mon, 25 Sep 2023 09:41:59 -0400 Subject: [PATCH] [docs] removed MaskFormerSwin and TimmBackbone from the table on index.md (#26347) removed MaskFormerSwin and TimmBackbone from the table --- docs/source/en/index.md | 2 -- utils/check_table.py | 5 +++++ 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/docs/source/en/index.md b/docs/source/en/index.md index cb1ab70fd4..4ca70fe251 100644 --- a/docs/source/en/index.md +++ b/docs/source/en/index.md @@ -393,7 +393,6 @@ Flax), PyTorch, and/or TensorFlow. | MarkupLM | ✅ | ❌ | ❌ | | Mask2Former | ✅ | ❌ | ❌ | | MaskFormer | ✅ | ❌ | ❌ | -| MaskFormerSwin | ❌ | ❌ | ❌ | | mBART | ✅ | ✅ | ✅ | | MEGA | ✅ | ❌ | ❌ | | Megatron-BERT | ✅ | ❌ | ❌ | @@ -462,7 +461,6 @@ Flax), PyTorch, and/or TensorFlow. | TAPAS | ✅ | ✅ | ❌ | | Time Series Transformer | ✅ | ❌ | ❌ | | TimeSformer | ✅ | ❌ | ❌ | -| TimmBackbone | ❌ | ❌ | ❌ | | Trajectory Transformer | ✅ | ❌ | ❌ | | Transformer-XL | ✅ | ✅ | ❌ | | TrOCR | ✅ | ❌ | ❌ | diff --git a/utils/check_table.py b/utils/check_table.py index 444036af99..5ebfca1fcd 100644 --- a/utils/check_table.py +++ b/utils/check_table.py @@ -173,7 +173,12 @@ def get_model_table_from_auto_modules() -> str: # Let's build that table! model_names = list(model_name_to_config.keys()) + + # MaskFormerSwin and TimmBackbone are backbones and so not meant to be loaded and used on their own. Instead, they define architectures which can be loaded using the AutoBackbone API. + names_to_exclude = ["MaskFormerSwin", "TimmBackbone"] + model_names = [name for name in model_names if name not in names_to_exclude] model_names.sort(key=str.lower) + columns = ["Model", "PyTorch support", "TensorFlow support", "Flax Support"] # We'll need widths to properly display everything in the center (+2 is to leave one extra space on each side). widths = [len(c) + 2 for c in columns]