amyeroberts
|
ba3264b4e8
|
Image Feature Extraction pipeline (#28216)
* Draft pipeline
* Fixup
* Fix docstrings
* Update doctest
* Update pipeline_model_mapping
* Update docstring
* Update tests
* Update src/transformers/pipelines/image_feature_extraction.py
Co-authored-by: Omar Sanseviero <osanseviero@gmail.com>
* Fix docstrings - review comments
* Remove pipeline mapping for composite vision models
* Add to pipeline tests
* Remove for flava (multimodal)
* safe pil import
* Add requirements for pipeline run
* Account for super slow efficientnet
* Review comments
* Fix tests
* Swap order of kwargs
* Use build_pipeline_init_args
* Add back FE pipeline for Vilt
* Include image_processor_kwargs in docstring
* Mark test as flaky
* Update TODO
* Update tests/pipelines/test_pipelines_image_feature_extraction.py
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
* Add license header
---------
Co-authored-by: Omar Sanseviero <osanseviero@gmail.com>
Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
|
2024-02-05 14:50:07 +00:00 |
|
NielsRogge
|
59499bbe8b
|
Update forward signature test for vision models (#27681)
* Update forward signature
* Empty-Commit
|
2023-11-27 15:48:17 +01:00 |
|
amyeroberts
|
c41291965f
|
🚨🚨🚨 Remove softmax for EfficientNetForImageClassification 🚨🚨🚨 (#25501)
* Remove softmax for EfficientNet
* Update integration test values
* Fix up
|
2023-08-14 17:08:47 +01:00 |
|
Sylvain Gugger
|
42571f6eb8
|
Make more test models smaller (#25005)
* Make more test models tiny
* Make more test models tiny
* More models
* More models
|
2023-07-24 10:08:47 -04:00 |
|
Sylvain Gugger
|
1023705440
|
Check models used for common tests are small (#24824)
* First models
* Conditional DETR
* Treat DETR models, skip others
* Skip LayoutLMv2 as well
* Fix last tests
|
2023-07-14 14:43:19 -04:00 |
|
Yih-Dar
|
c7ec71baf5
|
Update tiny model summary file for recent models (#22637)
* Update tiny model summary file for recent models
---------
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
|
2023-04-06 22:52:59 +02:00 |
|
Yih-Dar
|
fa01127a67
|
update_pip_test_mapping (#22606)
* Add TFBlipForConditionalGeneration
* update pipeline_model_mapping
* Add import
* Revert changes in GPTSanJapaneseTest
---------
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
|
2023-04-06 17:56:06 +02:00 |
|
Alara Dirik
|
49ab16239c
|
Add EfficientNet (#21563)
* Add EfficientNet to transformers
|
2023-02-20 16:37:11 +03:00 |
|