| import argparse | |
| import os | |
| import subprocess | |
| def batch_lora_json_extraction(src_dir, lora_json_extractor_path): | |
| for file in os.listdir(src_dir): | |
| if file.endswith(".safetensors"): | |
| input_file = os.path.join(src_dir, file) | |
| output_file = os.path.join(src_dir, file.replace(".safetensors", ".json")) | |
| subprocess.run(["python", lora_json_extractor_path, input_file, output_file]) | |
| if __name__ == "__main__": | |
| parser = argparse.ArgumentParser(description="Batch extraction of .safetensors files using lora_json_extractor.py.") | |
| parser.add_argument("src_dir", help="Source directory containing .safetensors files.") | |
| parser.add_argument("lora_json_extractor_path", help="Path to lora_json_extractor.py script.") | |
| args = parser.parse_args() | |
| batch_lora_json_extraction(args.src_dir, args.lora_json_extractor_path) | |