Add credit for stream_generator

This commit is contained in:
WeberJulian 2023-10-02 15:49:37 +02:00
parent a45cf83b34
commit 2ecf84a2c6
1 changed files with 2 additions and 0 deletions

View File

@ -1,3 +1,5 @@
# Adapted from: https://github.com/LowinLi/transformers-stream-generator
from transformers import ( from transformers import (
GenerationConfig, GenerationConfig,
GenerationMixin, GenerationMixin,