refactor code

Wanglongzhi2001 · Wanglongzhi2001 · commit 6ca08f7d91e4 · 2024-12-03T11:52:48.000+08:00
diff --git a/paddlenlp/experimental/transformers/generation_utils.py b/paddlenlp/experimental/transformers/generation_utils.py
@@ -792,7 +792,7 @@ def _post_process_(
                     model_kwargs["step_idx"],
                 )
 
-        speculate_method = self.config.speculate_method if hasattr(self.config, "speculate_method") else None
+        speculate_method = self.config.get("speculate_method", None)
         if speculate_method is not None:
             # Prepare output padding offset
             output_padding_offset, output_cum_offsets = self.get_output_padding_offset(

Original file line number	Diff line number	Diff line change
`@@ -792,7 +792,7 @@ def _post_process_(`
`792`	`792`	`model_kwargs["step_idx"],`
`793`	`793`	`)`
`794`	`794`
`795`		`- speculate_method = self.config.speculate_method if hasattr(self.config, "speculate_method") else None`
	`795`	`+ speculate_method = self.config.get("speculate_method", None)`
`796`	`796`	`if speculate_method is not None:`
`797`	`797`	`# Prepare output padding offset`
`798`	`798`	`output_padding_offset, output_cum_offsets = self.get_output_padding_offset(`