tests/test_pipelines_token_classification.py · e7f33e8cb38c446ce4004a742ebb386a4bef7174 · chenpangpang / transformers

Pass `model_kwargs` when loading a model in `pipeline()` (#12449) · e7f33e8c

Alex Hedges authored Jul 09, 2021

* Pass model_kwargs when loading a model in pipeline

* Add test for model_kwargs parameter of pipeline()

* Rewrite test to not download model

* Fix failing style checks

e7f33e8c

test_pipelines_token_classification.py 23.7 KB

Replace test_pipelines_token_classification.py