sparse / ms-swift /swift /cli /merge_lora.py
Enxin's picture
Upload folder using huggingface_hub
96fe658 verified
raw
history blame contribute delete
315 Bytes
# Copyright (c) Alibaba, Inc. and its affiliates.
from swift.llm import ExportArguments, SwiftPipeline, merge_lora
class SwiftMergeLoRA(SwiftPipeline):
args_class = ExportArguments
args: args_class
def run(self):
merge_lora(self.args)
if __name__ == '__main__':
SwiftMergeLoRA().main()