Add image classifier donut & update loss calculation for all swins (#37224)

* add classifier head to donut

* add to transformers __init__

* add to auto model

* fix typo

* add loss for image classification

* add checkpoint

* remove no needed import

* reoder import

* format

* consistency

* add test of classifier

* add doc

* try ignore

* update loss for all swin models
This commit is contained in:
AbdelKarim ELJANDOUBI
2025-04-10 15:00:42 +02:00
committed by GitHub
parent 5ae9b2cac0
commit 7ecc5b88c0
9 changed files with 177 additions and 48 deletions

View File

@@ -226,3 +226,8 @@ print(answer)
[[autodoc]] DonutSwinModel
- forward
## DonutSwinForImageClassification
[[autodoc]] transformers.DonutSwinForImageClassification
- forward