Fix: Correct hidden state return value in forward method

modeling.py

@@ -643,7 +643,7 @@ class Fast_dLLM_QwenForCausalLM(Fast_dLLM_QwenPreTrainedModel, GenerationMixin):
             loss=loss,
             logits=logits,
             past_key_values=outputs.past_key_values,
-            hidden_states=outputs.hidden_states,
+            hidden_states=outputs.last_hidden_state,
             attentions=outputs.attentions,
             block_past_key_values=outputs.block_past_key_values,
         )