Update modeling_auristream.py

modeling_auristream.py

@@ -357,11 +357,11 @@ class AuriStreamModel(AuriStreamPreTrainedModel):
         if not return_dict:
             if labels is not None:
                 return logits, loss
-            return logits.unsuqeeze(0), None
+            return logits.unsqueeze(0), None
         
         return CausalLMOutput(
             loss=loss,
-            logits=logits.unsuqeeze(0),
+            logits=logits.unsqueeze(0),
             hidden_states=all_hidden_states if output_hidden_states else None,
         )