diff --git a/presets/tuning/text-generation/cli.py b/presets/tuning/text-generation/cli.py index 4de48ad13..d6c0b0680 100644 --- a/presets/tuning/text-generation/cli.py +++ b/presets/tuning/text-generation/cli.py @@ -38,7 +38,7 @@ class ExtLoraConfig(LoraConfig): Lora Config """ init_lora_weights: bool = field(default=True, metadata={"help": "Enable initialization of LoRA weights"}) - target_modules: Optional[List[str]] = field(default=DEFAULT_TARGET_MODULES, metadata={"help": ("List of module names to replace with LoRA.")}) + target_modules: Optional[List[str]] = field(default_factory=lambda: DEFAULT_TARGET_MODULES if DEFAULT_TARGET_MODULES else None, metadata={"help": "List of module names to replace with LoRA."}) layers_to_transform: Optional[List[int]] = field(default=None, metadata={"help": "Layer indices to apply LoRA"}) layers_pattern: Optional[List[str]] = field(default=None, metadata={"help": "Pattern to match layers for LoRA"}) loftq_config: Dict[str, any] = field(default_factory=dict, metadata={"help": "LoftQ configuration for quantization"})