Upload 4 files

LyCORIS_mover/batch_lora_json_extractor.py

@@ -0,0 +1,18 @@
+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)

LyCORIS_mover/lora_json_extractor.py

@@ -0,0 +1,85 @@
+import json
+import sys
+
+
+# 既存の関数 (extract_metadata と save_metadata_to_json) は変更せずに再利用できます。
+
+def extract_metadata(_file_path, _start_marker):
+    local_metadata = ""
+
+    with open(_file_path, "rb") as f:
+        content = f.read()
+        start_pos = content.find(_start_marker)
+
+        if start_pos != -1:
+            end_pos = start_pos
+            brace_count = 0
+            for idx, b in enumerate(content[start_pos:]):
+                if b == ord('{'):
+                    brace_count += 1
+                elif b == ord('}'):
+                    brace_count -= 1
+
+                if brace_count == 0:
+                    end_pos += idx
+                    break
+
+            json_bytes = content[start_pos:end_pos + 1]
+            content_str = json_bytes.decode("utf-8")
+            try:
+                local_metadata = json.loads(content_str)
+            except json.JSONDecodeError as e:
+                print(f"Error decoding JSON: {e}")
+
+    return local_metadata
+
+
+def save_metadata_to_json(_metadata, _output_file):
+    with open(_output_file, "w", encoding="utf-8") as f:
+        json.dump(_metadata, f, ensure_ascii=False, indent=2)
+
+
+def extract_metadata_layer(json_data):
+    local_metadata_layer = {}
+
+    for key, value in json_data.items():
+        if key == "__metadata__":
+            local_metadata_layer[key] = value
+        elif isinstance(value, dict):
+            extracted = extract_metadata_layer(value)
+            if extracted:
+                local_metadata_layer[key] = extracted
+
+    return local_metadata_layer
+
+
+def load_json_file(_file_path):
+    with open(_file_path, "r", encoding="utf-8") as f:
+        return json.load(f)
+
+
+if __name__ == "__main__":
+    if len(sys.argv) != 3:
+        print("Usage: python module.py <input_file> <output_file>")
+        sys.exit(1)
+
+    file_path = sys.argv[1]
+    output_file = sys.argv[2]
+
+    # メタ情報の開始マーカー
+    start_marker = b'{"__metadata__":'
+
+    # メタ情報の抽出
+    metadata = extract_metadata(file_path, start_marker)
+
+    # メタ情報を整形して JSON ファイルに保存
+    save_metadata_to_json(metadata, output_file)
+
+    # JSON ファイルを読み込む
+    data = load_json_file(output_file)
+
+    # "__metadata__" 階層のみを抽出
+    metadata_layer = extract_metadata_layer(data)
+
+    # "__metadata__" 階層のみを含む JSON ファイルに保存
+    save_metadata_to_json(metadata_layer, output_file)

LyCORIS_mover/move_non_lora_safetensors.py

@@ -0,0 +1,27 @@
+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)

LyCORIS_mover/mover.bat

@@ -0,0 +1,22 @@
+chcp 65001
+@echo off
+echo "nannka mojibakesuru baainiha bat no senntouniaru chcp 65001 wo gyougoto sakujosuru"
+
+echo "loraフォルダのバックアップ取った?取ってないならそのままバツで閉じる。取ったならEnter"
+pause
+
+cd /D %~dp0
+echo "lora_dir_path にはloraディレクトリを指定"
+set lora_dir_path="H:\WEBUI_Last\models\Lora"
+
+echo "extractorはlora_json_extractorの場所を"
+set extractor=%~dp0\lora_json_extractor.py
+
+echo "lycoris_dir_pathには移動先となるリコリスのパスを指定"
+set lycoris_dir_path="H:\WEBUI_Last\models\LyCORIS"
+python batch_lora_json_extractor.py %lora_dir_path% %extractor%
+python move_non_lora_safetensors.py %lora_dir_path% %lycoris_dir_path%
+
+echo "移動完了！残ったjsonファイルは自力で削除してね(検索ボックスに.jsonで引っ掛けてCtrl+Aで選択してゴミ箱にでも移動させればOK)"
+echo "このウィンドウはバツ閉じすればOK"
+pause