Change streamer init argument name

Author: CodeWithKyrian

src/Generation/Streamers/Streamer.php

@@ -12,7 +12,7 @@
  */
 abstract class Streamer
 {
-    abstract public function init(PretrainedTokenizer $tokenizer, array $inputTokens, bool $includeInput): void;
+    abstract public function init(PretrainedTokenizer $tokenizer, array $inputTokens, bool $excludeInput = false): void;
 
     abstract public function put(mixed $value): void;
 

src/Generation/Streamers/TextStreamer.php

@@ -90,7 +90,7 @@ class TextStreamer extends Streamer
 {
     protected PretrainedTokenizer $tokenizer;
     protected array $inputTokens = [];
-    protected bool $includeInput = false;
+    protected bool $excludeInput = true;
     protected string $printedText = '';
     protected mixed $onStreamCallback = null;
     protected mixed $onStreamEndCallback = null;
@@ -108,13 +108,13 @@ public static function make(): self
         return new static();
     }
 
-    public function init(PretrainedTokenizer $tokenizer, array $inputTokens, bool $includeInput): void
+    public function init(PretrainedTokenizer $tokenizer, array $inputTokens, bool $excludeInput = false): void
     {
         $this->tokenizer = $tokenizer;
         $this->inputTokens = $inputTokens;
-        $this->includeInput = $includeInput;
+        $this->excludeInput = $excludeInput;
 
-        if (!$this->includeInput) {
+        if ($this->excludeInput) {
             $this->printedText = $this->tokenizer->decode($this->inputTokens, skipSpecialTokens: true);
             $this->printedLength = mb_strlen($this->printedText);
 

src/Pipelines/Text2TextGenerationPipeline.php

@@ -79,7 +79,7 @@ public function __invoke(array|string $inputs, ...$args): array
 
 
         // Streamer can only handle one input at a time for now, so we only pass the first input
-        $streamer->init($this->tokenizer, $inputIds[0]->toArray(), true);
+        $streamer->init($this->tokenizer, $inputIds[0]->toArray());
 
         // Generate output token ids
         $outputTokenIds = $this->model->generate($inputIds, generationConfig: $generateKwargs, streamer: $streamer);

src/Pipelines/TextGenerationPipeline.php

@@ -114,7 +114,7 @@ public function __invoke(array|string $inputs, ...$args): array
         );
 
         // Streamer can only handle one input at a time for now, so we only pass the first input
-        $streamer->init($this->tokenizer, $inputIds[0]->toArray(), false);
+        $streamer->init($this->tokenizer, $inputIds[0]->toArray(), true);
 
         $outputTokenIds = $this->model->generate($inputIds,
             generationConfig: $generationConfig,