transformers-stream-generator
Advanced tools
This is a text generation method which returns a generator, streaming out each token in real-time during inference, based on Huggingface/Transformers.
This is a text generation method which returns a generator, streaming out each token in real-time during inference, based on Huggingface/Transformers.
pip install transformers-stream-generator
from transformers_stream_generator import init_stream_support
init_stream_support()
do_stream=True in model.generate function and keep do_sample=True, then you can get a generatorgenerator = model.generate(input_ids, do_stream=True, do_sample=True)
for token in generator:
word = tokenizer.decode(token)
print(word)
FAQs
This is a text generation method which returns a generator, streaming out each token in real-time during inference, based on Huggingface/Transformers.
We found that transformers-stream-generator demonstrated a healthy version release cadence and project activity because the last version was released less than a year ago. It has 1 open source maintainer collaborating on the project.
Did you know?
Socket for GitHub automatically highlights issues in each pull request and monitors the health of all your open source dependencies. Discover the contents of your packages and block harmful activity before you install or update your dependencies.
Security News
PyPI confirms no security flaws were exploited in the Ultralytics supply chain attack and highlights improvements for safer package publishing.
Security News
Research
The Socket Research Team breaks down a malicious wrapper package that uses obfuscation to harvest credentials and exfiltrate sensitive data.
Research
Security News
Attackers used a malicious npm package typosquatting a popular ESLint plugin to steal sensitive data, execute commands, and exploit developer systems.