| import argparse | |
| import json | |
| import os | |
| import shutil | |
| def move_non_lora_safetensors(src_dir, dest_dir): | |
| for file in os.listdir(src_dir): | |
| if file.endswith(".json"): | |
| try: | |
| with open(os.path.join(src_dir, file), 'r', encoding='utf-8') as f: | |
| data = json.load(f) | |
| if data.get("__metadata__", {}).get("ss_network_module") != "networks.lora": | |
| safetensors_file = file.replace(".json", ".safetensors") | |
| src_path = os.path.join(src_dir, safetensors_file) | |
| dest_path = os.path.join(dest_dir, safetensors_file) | |
| if os.path.isfile(src_path): | |
| shutil.move(src_path, dest_path) | |
| except json.JSONDecodeError: | |
| continue | |
| if __name__ == "__main__": | |
| parser = argparse.ArgumentParser(description="Move .safetensors files of non-networks.lora JSON content.") | |
| parser.add_argument("src_dir", help="Source directory containing .json and .safetensors files.") | |
| parser.add_argument("dest_dir", help="Destination directory for non-networks.lora .safetensors files.") | |
| args = parser.parse_args() | |
| move_non_lora_safetensors(args.src_dir, args.dest_dir) | |