Update modeling_auristream.py

modeling_auristream.py

@@ -152,7 +152,7 @@ class AuriStream(PreTrainedModel):
                             logits=logits,
                             hidden_states=all_hidden_states,
                         )
-                     else:
+                    else:
                         model_output = CausalLMOutput(
                             loss=loss,
                             logits=logits,