generation_tf_logits_process.py
transformers
fb0ae129 - TF: XLA bad words logits processor and list of processors (#16974)
Commit
3 years ago
Author
Parents
- Files3
- src/transformers
- generation_tf_utils.py
- tests/generation
- test_generation_tf_logits_process.py