riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-03 12:50:06 +06:00
Code Issues Actions 28 Packages Projects Releases Wiki Activity

Export base streamer. (#36500)

Browse Source 
* Export base streamer. 

Previously, the base streamer class was not exported so the set of available streamers was fixed to 3 streamer classes. 

This change makes it so that customers may extend the default base streamer class.

* make fixup

---------

Co-authored-by: Joao Gante <joaofranciscocardosogante@gmail.com>
Co-authored-by: Joao Gante <joao@huggingface.co>
This commit is contained in:
Andreas Abdi 2025-03-07 03:16:09 -08:00 committed by GitHub
parent 6f775970c7
commit 8a16edce67
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/transformers/generation/__init__.py
View File

@ -26,7 +26,7 @@ _import_structure = {
"SynthIDTextWatermarkingConfig",
"WatermarkingConfig",
],
"streamers": ["AsyncTextIteratorStreamer", "TextIteratorStreamer", "TextStreamer"],
"streamers": ["AsyncTextIteratorStreamer", "BaseStreamer", "TextIteratorStreamer", "TextStreamer"],
}
try:
@ -197,7 +197,7 @@ if TYPE_CHECKING:
SynthIDTextWatermarkingConfig,
WatermarkingConfig,
)
from .streamers import AsyncTextIteratorStreamer, TextIteratorStreamer, TextStreamer
from .streamers import AsyncTextIteratorStreamer, BaseStreamer, TextIteratorStreamer, TextStreamer
try:
if not is_torch_available():