[hparam] add enable think argument (#7928)
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user