Making MaskFormerForInstanceSegmentation. (#15934)

Small adjustments.

Adding in type hint.

Last fix ?

Only include the default dict thing, not the pipelines.
This commit is contained in:
Nicolas Patry
2022-03-04 13:56:15 +01:00
committed by GitHub
parent 7ade7c1794
commit 89c7d9cfba
6 changed files with 42 additions and 0 deletions

View File

@@ -158,6 +158,10 @@ Likewise, if your `NewModel` is a subclass of [`PreTrainedModel`], make sure its
[[autodoc]] AutoModelForSemanticSegmentation
## AutoModelForInstanceSegmentation
[[autodoc]] AutoModelForInstanceSegmentation
## TFAutoModel
[[autodoc]] TFAutoModel