Update modeling_auristream.py

modeling_auristream.py

@@ -320,7 +320,7 @@ class AuriStream(PreTrainedModel):
 
         # First prediction of the model is the decoding of the last time bin
         logits = self.coch_head(x[:, [-1]])
-        predictions = [self.sample_logits(logits, temperature=temp)]
+        predictions = [self.sample_logits(logits, temperature=temp, top_k=top_k, top_p=top_p)]
         all_logits.append(logits)
 
         ### Predict future tokens