[docs] removed MaskFormerSwin and TimmBackbone from the table on index.md (#26347)

removed MaskFormerSwin and TimmBackbone from the table
This commit is contained in:
Maria Khalusova
2023-09-25 09:41:59 -04:00
committed by GitHub
parent 0ee4590684
commit 546e7679e7
2 changed files with 5 additions and 2 deletions

View File

@@ -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]