File size: 609 Bytes
cb2428f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
# Copyright (c) Alibaba, Inc. and its affiliates.
from dataclasses import dataclass
from swift.utils import get_logger
logger = get_logger()
@dataclass
class MergeArguments:
"""
MergeArguments is a dataclass that holds configuration for merging models.
Args:
merge_lora (bool): Flag to indicate if LoRA merging is enabled. Default is False.
safe_serialization(bool): Use safetensors or not, default `True`.
max_shard_size(str): The max size of single shard file.
"""
merge_lora: bool = False
safe_serialization: bool = True
max_shard_size: str = '5GB'
|