Update modeling_quiet.py

modeling_quiet.py

@@ -1697,8 +1697,8 @@ class QuietForCausalLM(QuietPreTrainedModel):
             hidden_states = outputs[0]
             prev_rm_logits = rm_logits  # for policy gradient
             prev_rm_tokens = cur_rm_tokens  # for policy gradient
-print("Hidden states contains NaN after loop:", torch.isnan(hidden_states).any().item())
-print("Logits contains NaN after loop:", torch.isnan(logits).any().item())
+            print("Hidden states contains NaN after loop:", torch.isnan(hidden_states).any().item())
+            print("Logits contains NaN after loop:", torch.isnan(logits).any().item())
             if ahead_idx == 0:
                 hidden_states_lm = hidden_states
                 logits = self.lm_head(hidden_states_lm)