This commit is contained in:
@@ -0,0 +1,15 @@
|
||||
# Copyright (c) ModelScope Contributors. All rights reserved.
|
||||
from swift.arguments import ExportArguments
|
||||
from swift.pipelines import SwiftPipeline, merge_lora
|
||||
|
||||
|
||||
class SwiftMergeLoRA(SwiftPipeline):
|
||||
args_class = ExportArguments
|
||||
args: args_class
|
||||
|
||||
def run(self):
|
||||
merge_lora(self.args)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
SwiftMergeLoRA().main()
|
||||
Reference in New Issue
Block a user