Commit 19470842 authored by zhuwenwen's avatar zhuwenwen
Browse files

update arg_utils.py

parent 7f022e4d
......@@ -706,9 +706,7 @@ class EngineArgs:
lora_group.add_argument('--max-lora-rank',
**lora_kwargs["max_lora_rank"])
lora_group.add_argument('--merge-lora',
**lora_kwargs["merge-lora"])
# action=argparse.BooleanOptionalAction,
# help='If set to True, the weights of the base layer will be merged with the weights of Lora.')
**lora_kwargs["merge_lora"])
lora_group.add_argument('--lora-target-modules',
**lora_kwargs["lora_target_modules"])
lora_group.add_argument('--lora-extra-vocab-size',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment