423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-15 16:18:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update generating_args.py

Browse Source 
Former-commit-id: 861c146fa7d9cb5b99372464bd068c20fa36415d
This commit is contained in:
hoshi-hiyouga 2024-05-20 00:29:31 +08:00 committed by GitHub
parent 17d398f419
commit 3578abc7a4
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/llamafactory/hparams/generating_args.py
View File

@ -1,5 +1,5 @@
from dataclasses import asdict, dataclass, field from dataclasses import asdict, dataclass, field
from typing import Any, Dict from typing import Any, Dict, Optional
@dataclass @dataclass
@ -46,10 +46,9 @@ class GeneratingArguments:
default=1.0, default=1.0,
metadata={"help": "Exponential penalty to the length that is used with beam-based generation."}, metadata={"help": "Exponential penalty to the length that is used with beam-based generation."},
) )
system_message: str = field( default_system: Optional[str] = field(
default=None, default=None,
metadata={ metadata={"help": "Default system message to use in chat completion."},
"help": "System message is a message that the developer wrote to tell the bot how to interpret the conversation"},
) )
def to_dict(self) -> Dict[str, Any]: def to_dict(self) -> Dict[str, Any]: