Skip to content

Commit 8c841d4

Browse files
fix typing (#974)
1 parent 9b08735 commit 8c841d4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

swift/llm/utils/argument.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -462,13 +462,13 @@ class SftArguments(ArgumentsBase):
462462
boft_block_size: int = 4
463463
boft_block_num: int = 0
464464
boft_n_butterfly_factor: int = 1
465-
boft_target_modules: Optional[Union[List[str], str]] = field(default_factory=lambda: ['DEFAULT'])
465+
boft_target_modules: List[str] = field(default_factory=lambda: ['DEFAULT'])
466466
boft_dropout: float = 0.0
467467
boft_modules_to_save: List[str] = field(default_factory=list)
468468

469469
# Vera
470470
vera_rank: int = 256
471-
vera_target_modules: Optional[Union[List[str], str]] = field(default_factory=lambda: ['DEFAULT'])
471+
vera_target_modules: List[str] = field(default_factory=lambda: ['DEFAULT'])
472472
vera_projection_prng_key: int = 0
473473
vera_dropout: float = 0.0
474474
vera_d_initial: float = 0.1

0 commit comments

Comments
 (0)