Lora Sorter (python checkpoint type inspection tool)
详情
下载文件
关于此版本
模型描述
这是一对用于检测 Lora 基础模型类型并将 Lora 文件移动到相应文件夹的 Python 脚本。
如果你有 1000 多个 Lora 文件放在同一个文件夹中，却不确定哪些是 SD、哪些是 SDXL、哪些是 Flux 等，这个 Python 程序会分析所有文件，生成一个计划，并在你执行前提供机会检查和修正其结果。
共有三个基本步骤。
扫描文件夹，分析文件，生成第一个 CSV（电子表格）文件
读取 CSV 文件，确定每个文件的去向，并向用户展示计划
执行计划
注意：在步骤 2 和步骤 3 之间，你应该打开并审查/修正 CSV 文件的内容。
基本命令：
python inspect_lora.py "X:\路径\包含\lora\safetensor\文件\的文件夹\"
python inspect_sort_lora.py lora_inspect_index.csv
python inspect_sort_lora.py lora_inspect_index.csv --base "..." --apply
我建议你在仔细查看第二个 CSV 文件（lora_inspect_index_sort_plan.csv）的内容之前，不要运行 --apply，因为这个程序有时会出错，尤其容易混淆较旧的 SD 和 SDXL Lora，或那些未被正确生成的文件。
三步流程的目的，正是为了给你机会修正它的错误。
第一部分/二 - inspect_lora.py
- 该脚本将分析所有目标 Lora 文件，并生成一个 CSV 文件。
生成的 CSV 列（部分列将在下一步填充）：
- {路径家族, 总张量数, 含有flux块, 含有输入块, 含有下采样块, 含有双文本编码器, 错误, 可能为flux, 可能为sdxl, 可能为sd, 可能为lcm_turbo, 可能为ltxv, 可能为pony, 可能为ponyxl, 可能为qwen, 可能为wan, 移动目标}
CLI 使用方式（命令行选项）：
python inspect_lora.py "D:\AI\stable-diffusion-webui\models\Lora\some_lora.safetensors"
python inspect_lora.py "D:\AI\stable-diffusion-webui\models\Lora\*.safetensors"
python inspect_lora.py "D:\AI\stable-diffusion-webui\models\Lora\SD\"
第二部分/二 - inspect_sort_lora.py
此脚本读取 lora_inspect_index.csv，并根据家族对 LoRA 进行分类（或直接应用分类）。它专为配合 inspect_lora.py 的输出而设计。
该脚本：
加载该 CSV 文件
对于每个路径有效且无错误的行，判断该 LoRA 应归属的“类别”（例如：“SDXL”、“SD15”、“Flux”、“未知”）
可选地，根据该类别将文件移动到子文件夹中
总是生成第二个 CSV 文件，包含计划/结果：lora_sort_plan.csv
CLI 使用方式（命令行选项）：
python inspect_sort_lora.py lora_inspect_index.csv
python inspect_sort_lora.py lora_inspect_index.csv --base "D:\AI\stable-diffusion-webui\models\Lora"
python inspect_sort_lora.py lora_inspect_index.csv --base "..." --apply