From a49ebd2ab9667d662de0542bf888789b91c1559f Mon Sep 17 00:00:00 2001 From: padluo <454316879@qq.com> Date: Thu, 14 Mar 2024 01:35:41 +0000 Subject: [PATCH] [bugfix][Pytorch][Telechat] fix SyntaxError Signed-off-by: padluo <454316879@qq.com> --- PyTorch/contrib/nlp/Telechat/modeling_telechat.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/PyTorch/contrib/nlp/Telechat/modeling_telechat.py b/PyTorch/contrib/nlp/Telechat/modeling_telechat.py index 4704595634..7a96a67257 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 -- Gitee