File size: 315 Bytes
cb2428f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
# 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()
|