Update modeling_quiet.py

modeling_quiet.py

@@ -1121,6 +1121,7 @@ class QuietModel(QuietPreTrainedModel):
                         # Truncate the attention mask to match the sequence length
                         attention_mask = attention_mask[:, :, :, :seq_length]
 
+            # Assign the value to hidden_states after the attention mask preparation
             hidden_states = inputs_embeds
 
         # decoder layers