transformers/tests/models/timm_wrapper
Pavel Iakubovskii 59e5b3f01b
Timm wrapper label names (#35553)
* Add timm wrapper label names mapping

* Add index to classification pipeline

* Revert adding index for pipelines

* Add custom model check for loading timm labels

* Add tests for labels

* [run-slow] timm_wrapper

* Add note regarding label2id mapping
2025-01-08 14:09:46 +00:00
..
__init__.py Add TimmWrapper (#34564) 2024-12-11 12:40:30 +00:00
test_image_processing_timm_wrapper.py Add TimmWrapper (#34564) 2024-12-11 12:40:30 +00:00
test_modeling_timm_wrapper.py Timm wrapper label names (#35553) 2025-01-08 14:09:46 +00:00