Lora Sorter (python checkpoint type inspection tool)

이는 LoRA 기본 모델 유형을 감지하고 LoRA를 적절한 폴더로 이동시키는 파이썬 스크립트 쌍입니다.

1000개 이상의 LoRA가 같은 폴더에 있고, 어떤 것이 SD인지, SDXL인지, Flux인지 확실하지 않다면, 이 파이썬 프로그램은 모든 파일을 분석하여 계획을 생성하고, 계획을 실행하기 전에 그 결과를 확인하고 수정할 기회를 제공합니다.

세 가지 기본 단계가 있습니다.

1. 폴더를 스캔하고 파일을 분석하여 첫 번째 CSV(스프레드시트) 파일을 생성합니다

2. CSV 파일을 읽고, 각 파일의 이동 위치를 결정한 후 사용자에게 계획을 표시합니다

3. 계획을 실행합니다

참고: 단계 2와 3 사이에는 CSV 파일의 내용을 열어 검토하고 수정해야 합니다.

기본 명령어:

1. python inspect_lora.py "X:\lora_safetensor_파일이_있는_경로\"

2. python inspect_sort_lora.py lora_inspect_index.csv

3. python inspect_sort_lora.py lora_inspect_index.csv --base "..." --apply

두 번째 CSV 파일(lora_inspect_index_sort_plan.csv)의 내용을 충분히 검토한 후에야 "--apply"를 실행하길 권장합니다. 이 도구는 일부 오류를 일으킬 수 있으며, 오래된 SD와 SDXL LoRA 또는 정확히 생성되지 않은 LoRA를 혼동하기 때문입니다.

이 세 단계 과정의 목적은 프로그램의 오류를 수정할 기회를 제공하는 것입니다.

PART 1/2 - inspect_lora.py

• 이 스크립트는 대상 LoRA 파일을 모두 분석하여 CSV 파일을 생성합니다.

생성되는 CSV 열: (다음 단계에서만 채워지는 항목도 있음)

• {path family, total_tensors, has_flux_blocks, has_input_blocks, has_down_blocks, has_dual_te, error, maybe_flux, maybe_sdxl, maybe_sd, maybe_lcm_turbo, maybe_ltxv, maybe_pony, maybe_ponyxl, maybe_qwen, maybe_wan, move destination}

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\"

PART 2/2 - inspect_sort_lora.py

이 스크립트는 lora_inspect_index.csv를 읽고 LoRA를 유형별로 정렬하는 계획을 제안하거나 적용합니다. inspect_lora.py의 출력과 함께 작동하도록 설계되었습니다.

이 스크립트는...

• 해당 CSV를 로드합니다

• 유효한 경로를 가지며 오류가 없는 각 행에 대해 LoRA가 속하는 "버킷"(예: "SDXL", "SD15", "Flux", "Unknown")을 결정합니다

• 선택적으로 해당 버킷에 따라 파일을 하위 폴더로 이동합니다

• 항상 계획/결과를 담은 두 번째 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