Add TFAutoModelForSemanticSegmentation to the main __init__.py (#18600)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2022-08-12 15:10:00 +02:00
committed by GitHub
parent 4eed2beca0
commit 2156619f10
4 changed files with 15 additions and 0 deletions

View File

@@ -186,6 +186,10 @@ Likewise, if your `NewModel` is a subclass of [`PreTrainedModel`], make sure its
[[autodoc]] TFAutoModelForImageClassification
## TFAutoModelForSemanticSegmentation
[[autodoc]] TFAutoModelForSemanticSegmentation
## TFAutoModelForMaskedLM
[[autodoc]] TFAutoModelForMaskedLM