diff --git a/PyTorch/contrib/nlp/Telechat/modeling_telechat.py b/PyTorch/contrib/nlp/Telechat/modeling_telechat.py
index 47045956345abfd695eb1f4c5024797eddeb67bb..7a96a67257528ea028b46dd53d3dab68630f60b9 100644
--- a/PyTorch/contrib/nlp/Telechat/modeling_telechat.py
+++ b/PyTorch/contrib/nlp/Telechat/modeling_telechat.py
@@ -628,8 +628,10 @@ class TelechatModel(TelechatPreTrainedModel):
             **deprecated_arguments,
     ) -> Union[Tuple[torch.Tensor, ...], BaseModelOutputWithPastAndCrossAttentions]:
 
-        output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions        output_hidden_states = (
-            output_hidden_states if output_hidden_states is not None else self.config.output_hidden_states              )
+        output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
+        output_hidden_states = (
+            output_hidden_states if output_hidden_states is not None else self.config.output_hidden_states
+        )
         use_cache = use_cache if use_cache is not None else self.config.use_cache
         return_dict = return_dict if return_dict is not None else self.config.use_return_dict