Update modeling_quiet.py

modeling_quiet.py

@@ -1704,11 +1704,10 @@ class QuietForCausalLM(QuietPreTrainedModel):
             hidden_states = outputs[0]
             prev_rm_logits = rm_logits  # for policy gradient
             prev_rm_tokens = cur_rm_tokens  # for policy gradient
-
             if ahead_idx == 0:
                 hidden_states_lm = hidden_states
-				logits = self.lm_head(hidden_states_lm)
-				logits = self.bn_lm_head(logits.transpose(1, 2)).transpose(1, 2)
+                logits = self.lm_head(hidden_states_lm)
+                logits = self.bn_lm_head(logits.transpose(1, 2)).transpose(1, 2)
                 base_hidden_states = hidden_states.clone()
                 initial_loss_logits = logits.clone()
                 if self.optimize_lm_head_only_at_start or self.optimize_model_only_at_start: