SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/en/index.md
View File

@@ -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 | ✅ | ❌ | ❌ |

5
utils/check_table.py
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]