Merge pull request #22 from CodeWithKyrian/fix-streamer-init

bugfix: Fix error that occurs when streamer is not used

Author: CodeWithKyrian

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());
+        $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(), true);
+        $streamer?->init($this->tokenizer, $inputIds[0]->toArray(), true);
 
         $outputTokenIds = $this->model->generate($inputIds,
             generationConfig: $generationConfig,