tests/pipelines/test_pipelines_text_generation.py · 2ecefc3959705b3b4f696736fa1c5b44102f31c7 · chenpangpang / transformers

Add chat templating support for KeyDataset in text-generation pipeline (#30558) · 2ecefc39

DarshanDeshpande authored Apr 30, 2024

* added chat templating support for keydataset in generation pipeline

* fixed and improved test

* fix formatting test failures

* Fix tests

* Fix tests

2ecefc39

test_pipelines_text_generation.py 19.3 KB

Replace test_pipelines_text_generation.py