Update modeling_quiet.py

modeling_quiet.py

@@ -1674,9 +1674,9 @@ class QuietForCausalLM(QuietPreTrainedModel):
                         num_thought_tokens += 1
                     if self.use_end_thought_token:
                         num_thought_tokens += 1
-                        
+                       
+                    original_sequence_length = input_ids.shape[1]    
                     seq_length = original_sequence_length + num_thought_tokens
-                    original_sequence_length = input_ids.shape[1]
                     # # if the attention mask 
                     attention_mask = _prepare_4d_causal_attention_mask(
                         attention_mask,