diff --git a/src/Models/Pretrained/PretrainedModel.php b/src/Models/Pretrained/PretrainedModel.php
index d67f2e1..4188581 100644
--- a/src/Models/Pretrained/PretrainedModel.php
+++ b/src/Models/Pretrained/PretrainedModel.php
@@ -426,7 +426,7 @@ public function generate(
 
             $streamer?->put($generatedInputIds);
 
-            $stop = $stoppingCriteria($generatedInputIds, $scores);
+            $stop = $stoppingCriteria($allInputIds, $scores);
             if (array_every($stop, fn($x) => $x)) {
                 break;
             }