diff --git a/src/lmflow/args.py b/src/lmflow/args.py index 1f80ece1d..4d6199db4 100644 --- a/src/lmflow/args.py +++ b/src/lmflow/args.py @@ -591,7 +591,7 @@ class FinetunerArguments(TrainingArguments): "help": "the number of steps in each freezing interval of LISA, i.e. the selected unfreezed layers are randomly switched every {lisa_interval_steps} steps." } ) - lisa_layers_attribute: int = field( + lisa_layers_attribute: str = field( default="model.model.layers", metadata={ "help": "where the layer attribute stores, e.g. model.model.layers"