lab_settings package

Submodules

Settings manager.

class config.lab_settings.SFTParams(max_length: int, batch_size: int, max_fine_tuning_steps: int, device: str, finetuned_model_path: Path, learning_rate: float, target_modules: list[str] | None = None)

Bases: object

Fine-tuning parameters.

__init__(*args: Any, **kwargs: Any) → None

batch_size: int

device: str

finetuned_model_path: Path

learning_rate: float

max_fine_tuning_steps: int

max_length: int

target_modules: list[str] | None = None