LoRAとLyCORISファイルを分類するスクリプトを作った
Stable Diffusion を使っていて「この.safetensorsはLoRAなのかLyCORISなのかわからん…」 「LoRA/LyCORISでフォルダを分けて運用することを推奨されてる気がするけど無理じゃない?」 と悩むことに疲れたので判別するPythonスクリプトを作った。
以下説明。
Lora Lyco 分類器
`.safetensors`ファイルが"LORA"か"LyCORIS"かを判別し各ディレクトリに移動するシンプルなスクリプトです。Stable-Diffusion-Webui-Civitai-Helper extension により生成される `.civitai.info` から判別します。
使い方(Pythonスクリプト実行)
`lora_lyco_classifier.py`を`.civitai.info`ファイルとその他の関連ファイル(`.safetensors`や`.preview.png`)を含むディレクトリに配置してください。
Python 3を使用してスクリプトを実行します。追加の引数は必要ありません
macOS または Linux:
python3 lora_lyco_classifier.py
Windows:
py lora_lyco_classifier.py
またはスクリプトを直接実行:
.\lora_lyco_classifier.py
スクリプトを実行すると以下の工程を経て'_lora'と'_lycoris'ディレクトリにファイルが移動します。
スクリプトと同じディレクトリ内のすべてのファイルをスキャンします。
`.civitai.info`ファイルの内容から'model type'が'LORA'または'LyCORIS'のいずれであるか判別します。
'_lora' および '_lycoris' ディレクトリを作成します。
'model type'に基づき'LOAR', 'LyCORIS'ファイルをそれぞれのディレクトリに移動します。
使い方(Windows: exeファイル実行)
スクリプトの実行が面倒なので実行ファイルも作りました。
lora_lyco_classifier.exeを.civitai.infoファイルとその他の関連ファイル(.safetensorsや.preview.png)を含むディレクトリに配置して実行すれば上記と同じ挙動をします。