Patrick von Platen
48fa42e5d5
Add Speech AutoModels ( #13655 )
...
* upload
* correct
* correct
* correct
* finish
* up
* up
* up again
2021-09-21 08:50:33 +02:00
Mishig Davaadorj
2a15e8ccfb
Object detection pipeline ( #12886 )
...
* Implement object-detection pipeline
* Define threshold const
* Add `threshold` argument
* Refactor
* Uncomment test inputs
* `rm
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Fix typo
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Fix typo
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Chore better doc
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Rm unnecessary lines
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Chore better naming
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Fix typo
* Add `detr-tiny` for tests
* Add `ObjectDetectionPipeline` to `trnsfrmrs/init`
* Implement new bbox format
* Update detr post_process
* Update `load_img` method obj det pipeline
* make style
* Implement new testing format for obj det pipeln
* Add guard pytorch specific code in pipeline
* Add doc
* Make pipeline_obj_tet tests deterministic
* Revert some changes to `post_process` COCO api
* Chore
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Update src/transformers/pipelines/object_detection.py
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
* Rm timm requirement
* make fixup
* Add timm requirement to test
* Make fixup
* Guard torch.Tensor
* Chore
* Delete unnecessary comment
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>
2021-09-08 17:17:32 +02:00
Anton Lozhkov
b9c6a97694
Add the AudioClassificationPipeline ( #13342 )
...
* Add the audio classification pipeline
* Remove autoconfig exception
* Mark ffmpeg test as slow
* Rearrange pipeline tests
* Add small test
* Replace asserts with ValueError
2021-09-01 11:03:48 +03:00
Suraj Patil
1498eb9888
add FlaxAutoModelForImageClassification in main init ( #12298 )
2021-06-22 18:26:05 +05:30
Suraj Patil
f74655cd9b
[Flax] FlaxAutoModelForSeq2SeqLM ( #12228 )
...
* add FlaxAutoModelForSeq2SeqLM
2021-06-18 13:20:09 +05:30
Suraj Patil
ca33278fdb
FlaxGPT2 ( #11556 )
...
* flax gpt2
* combine masks
* handle shared embeds
* add causal LM sample
* style
* add tests
* style
* fix imports, docs, quality
* don't use cache
* add cache
* add cache 1st version
* make use cache work
* start adding test for generation
* finish generation loop compilation
* rewrite test
* finish
* update
* update
* apply sylvains suggestions
* update
* refactor
* fix typo
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2021-05-18 22:50:51 +01:00
Lysandre Debut
39084ca663
Add the ImageClassificationPipeline ( #11598 )
...
* Add the ImageClassificationPipeline
* Code review
Co-authored-by: patrickvonplaten <patrick.v.platen@gmail.com>
* Have `load_image` at the module level
Co-authored-by: patrickvonplaten <patrick.v.platen@gmail.com>
2021-05-07 08:08:40 -04:00
Sylvain Gugger
403d530eec
Auto feature extractor ( #11097 )
...
* AutoFeatureExtractor
* Init and first tests
* Tests
* Damn you gitignore
* Quality
* Defensive test for when not all backends are here
* Use pattern for Speech2Text models
2021-04-06 19:20:08 -04:00
Sylvain Gugger
6c25f5228e
Refactor AutoModel classes and add Flax Auto classes ( #11027 )
...
* Refactor AutoModel classes and add Flax Auto classes
* Add new objects to the init
* Fix hubconf and sort models
* Fix TF tests
* Missing coma
* Update src/transformers/models/auto/auto_factory.py
Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
* Fix init
* Fix dummies
* Other init to fix
Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
2021-04-05 10:11:28 -04:00
Lysandre Debut
07384baf7a
AutoModelForTableQuestionAnswering ( #9154 )
...
* AutoModelForTableQuestionAnswering
* Update src/transformers/models/auto/modeling_auto.py
* Style
2020-12-16 12:14:33 -05:00
Sylvain Gugger
1310e1a758
Enforce all objects in the main init are documented ( #9014 )
2020-12-10 11:57:12 -05:00
Sylvain Gugger
00aa9dbca2
Copyright ( #8970 )
...
* Add copyright everywhere missing
* Style
2020-12-07 18:36:34 -05:00
Patrick von Platen
9c83b96e62
[Tests] Add Common Test for Training + Fix a couple of bugs ( #8415 )
...
* add training tests
* correct longformer
* fix docs
* fix some tests
* fix some more train tests
* remove ipdb
* fix multiple edge case model training
* fix funnel and prophetnet
* clean gpt models
* undo renaming of albert
2020-11-09 18:24:41 +01:00
Sylvain Gugger
6241c873cd
Document the various LM Auto models ( #8118 )
2020-10-28 13:41:56 -04:00
Sylvain Gugger
08f534d2da
Doc styling ( #8067 )
...
* Important files
* Styling them all
* Revert "Styling them all"
This reverts commit 7d029395fd
2020-10-26 18:26:02 -04:00
Sylvain Gugger
3323146e90
Models doc ( #7345 )
...
* Clean up model documentation
* Formatting
* Preparation work
* Long lines
* Main work on rst files
* Cleanup all config files
* Syntax fix
* Clean all tokenizers
* Work on first models
* Models beginning
* FaluBERT
* All PyTorch models
* All models
* Long lines again
* Fixes
* More fixes
* Update docs/source/model_doc/bert.rst
Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
* Update docs/source/model_doc/electra.rst
Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
* Last fixes
Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
2020-09-23 13:20:45 -04:00
sgugger
5636cbb25d
Extra )
2020-09-14 09:37:55 -04:00
Sylvain Gugger
ccc8e30c8a
Clean up autoclass doc ( #7081 )
2020-09-14 09:26:41 -04:00
Sylvain Gugger
1262495a91
Add TF auto model to the docs + fix sphinx warnings ( #5187 )
2020-06-22 14:43:52 -04:00
Sylvain Gugger
011cc0be51
Fix all sphynx warnings ( #5068 )
2020-06-16 16:50:02 -04:00
thomwolf
e0849a66ac
adding in the doc
2020-01-27 14:27:07 -05:00
Lysandre
983fef469c
AutoModels doc
2020-01-24 16:37:30 -05:00
alberduris
81d6841b4b
GPU text generation: mMoved the encoded_prompt to correct device
2020-01-06 15:11:12 +01:00
alberduris
dd4df80f0b
Moved the encoded_prompts to correct device
2020-01-06 15:11:12 +01:00
thomwolf
31c23bd5ee
[BIG] pytorch-transformers => transformers
2019-09-26 10:15:53 +02:00
thomwolf
0b524b0848
remove derived classes for now
2019-08-05 19:08:19 +02:00
thomwolf
13936a9621
update doc and tests
2019-08-05 18:48:16 +02:00
thomwolf
b90e29d52c
working on automodels
2019-08-05 16:06:34 +02:00
