423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-08-23 22:32:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update generating_args.py

Browse Source 
Former-commit-id: 7ae7ae64f0e9e8661f9efd30997f8b96673d467a
This commit is contained in:
hoshi-hiyouga 2024-05-07 00:27:56 +08:00 committed by GitHub
parent 189346188b
commit d65b2332cf
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/llmtuner/hparams/generating_args.py
View File

@ -1,5 +1,5 @@
from dataclasses import asdict, dataclass, field
from typing import Any, Dict, Union, Optional, List
from typing import Any, Dict
@dataclass
@ -46,10 +46,6 @@ class GeneratingArguments:
default=1.0,
metadata={"help": "Exponential penalty to the length that is used with beam-based generation."},
)
stop: Union[Optional[str], List[str]] = field(
default=None,
metadata={"help": "List of strings or string that stop the generation when they are generated. The returned output will not contain the stop strings."},
)
def to_dict(self) -> Dict[str, Any]:
args = asdict(self)
if args.get("max_new_tokens", -1) > 0: