11 lines
248 B
Python
11 lines
248 B
Python
# Copyright (c) ModelScope Contributors. All rights reserved.
|
|
from dataclasses import dataclass
|
|
|
|
from .sft_args import SftArguments
|
|
|
|
|
|
@dataclass
|
|
class PretrainArguments(SftArguments):
|
|
use_chat_template: bool = False
|
|
loss_scale: str = 'all'
|