SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Generate: logits processors are doctested and fix broken doctests (#25692)

Browse Source 
* shorter example

* add logits processors to doctests

* remove file from conflict?

* tmp commit

* Fix broken tests; Shorter sampling tests

* Apply suggestions from code review

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>

---------

Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>
This commit is contained in:
Joao Gante
2023-08-25 12:42:06 +01:00
committed by GitHub
parent c6a84b7202
commit 494e96d8d6
2 changed files with 93 additions and 150 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
utils/not_doctested.txt
View File

@@ -371,7 +371,6 @@ src/transformers/generation/beam_constraints.py
src/transformers/generation/beam_search.py
src/transformers/generation/flax_logits_process.py
src/transformers/generation/flax_utils.py
src/transformers/generation/logits_process.py
src/transformers/generation/stopping_criteria.py
src/transformers/generation/streamers.py
src/transformers/generation/tf_logits_process.py