from pathlib import Path

from src.settings.base import ExtraFieldsNotAllowedBaseModel
from src.settings.model import PreTrainedModelSettings
from src.settings.tf.tokenizer import TokenizerSettings


class MergeSettings(ExtraFieldsNotAllowedBaseModel):
    tokenizer_settings: TokenizerSettings
    old_model_settings: PreTrainedModelSettings
    cur_model_settings: PreTrainedModelSettings

    alpha: float

    save_path: Path
