Unverified Commit da7059b5 authored by josephrocca's avatar josephrocca Committed by GitHub
Browse files

fix: pass `args.output_path` to `compose_lora` when called via command line (#364)

* Add missing `save_file` call for use of composed.py via command line

* use `output_path` functionality of `compose_lora` instead
parent 2eb3fe1e
...@@ -136,4 +136,4 @@ if __name__ == "__main__": ...@@ -136,4 +136,4 @@ if __name__ == "__main__":
parser.add_argument("-o", "--output-path", type=str, required=True, help="path to the output safetensors file") parser.add_argument("-o", "--output-path", type=str, required=True, help="path to the output safetensors file")
args = parser.parse_args() args = parser.parse_args()
assert len(args.input_paths) == len(args.strengths) assert len(args.input_paths) == len(args.strengths)
composed = compose_lora(list(zip(args.input_paths, args.strengths))) compose_lora(list(zip(args.input_paths, args.strengths)), args.output_path)
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