[hparam] add enable think argument (#7928)

This commit is contained in:
hoshi-hiyouga
2025-04-30 17:21:30 +08:00
committed by GitHub
parent c566e39b7d
commit 13b05e74f1
4 changed files with 10 additions and 3 deletions

View File

@@ -70,6 +70,10 @@ class GeneratingArguments:
default=True,
metadata={"help": "Whether or not to remove special tokens in the decoding."},
)
enable_thinking: bool = field(
default=True,
metadata={"help": "Whether or not to enable thinking mode for reasoning models."},
)
def to_dict(self, obey_generation_config: bool = False) -> dict[str, Any]:
args = asdict(self)