LoRA는 고통스러울 수 있다!

_safetriggers.py_는 태그된 소스 이미지로 학습된 LoRA에서 TAG, TRIGGER-WORD 및 TRIGGER-PHRASE를 추출하여 화면에 표시하거나 다음 형태의 파일로 저장할 수 있는 Linux용 CLI Python 스크립트입니다(Ubuntu Server 24.10에서 테스트됨).

<LoRA 이름>.safetriggers

이 파일은 LoRA와 함께 사용자에게 빠른 참조용으로 제공되며, 작고 간단한 JSON 텍스트 파일입니다.

참고. LoRA(*.safetensors)는 영향을 받지 않으며 오직 읽기 전용입니다.

트리거는 소스 이미지에서 태그가 발생한 순서(학습된 순서)에 따라 가장 중요한 항목이 먼저 오도록 정렬됩니다.

이렇게 하면 LoRA가 실제로 무엇에 대해 학습되었는지 쉽게 파악할 수 있으며, 때로는 놀라운 결과를 얻을 수 있습니다.

Q. LoRA 사용자! "foo_bar.safetensors"라는 LoRA의 트리거가 무엇인지 쉽게 확인하고 싶으신가요?

python3 safetriggers.py --lora foo_bar

Q. LoRA 제작자! "foo_bar.safetensors"라는 LoRA가 작동하기 위해 어떤 트리거가 필요한지 사용자에게 알려주는 파일을 쉽게 생성하고 싶으신가요?

python3 safetriggers.py --mksafetriggers --lora foo_bar

Q. LoRA 제작자! "./loras/foo_bar.safetensors"에 있는 LoRA가 어떤 태그로 학습되었으며, 소스 이미지에서 어떤 내용을 중요한 것으로 인식했는지 쉽게 확인하고 싶으신가요?

python3 safetriggers.py --mksafetriggers --lorapath ./loras --lora foo_bar

그러면 ./loras/foo_bar.safetriggers 파일이 생성되어 해당 정보를 처리, 읽기, 검토하거나 .safetensors 파일과 함께 배포할 수 있습니다.

2025년 3월 11일 이후 버전:

새로운 -allloras 옵션으로 인해 더 이상 _-lora_를 사용하여 개별 LoRA를 지정할 필요가 없습니다. -lorapath 및 하위 경로에 있는 모든 LoRA에 대해 .safetriggers 파일을 검토하고 생성하려면:

python3 safetriggers.py --mksafetriggers --allloras --lorapath ./loras/nsfw

2025년 3월 13일 이후 버전:

새로운 -safetriggers 옵션을 사용하여 개별 LoRA의 .safetriggers 파일을 지정하고 그 내용을 화면에 표시할 수 있습니다. 예:

python3 safetriggers.py --lorapath ./loras/nsfw --safetriggers PrettyWomen-f1-v8

이 명령은 LoRA가 학습한 정렬된 태그, 트리거 단어, 트리거 문구를 화면에 표시합니다.

2025년 3월 16일 이후 버전:

새로운 옵션을 사용하여 개별 LoRA의 .safetriggers 파일을 지정하고 그 내용을 화면에 표시할 수 있습니다. 예:

python3 safetriggers.py -p ./loras/nsfw -s PrettyWomen-f1-v8

이 명령은 LoRA가 학습한 정렬된 태그, 트리거 단어, 트리거 문구를 화면에 표시합니다.

이러한 짧은 CLI 매개변수는 다음 명령어로 화면에 표시됩니다:

python3 safetriggers.py -h

참고.

일부 LoRA는 태그를 포함하지 않으며(그러한 LoRA는 safetriggers가 식별 가능), 따라서 읽을 내용이 없으므로 유일한 선택지는 LoRA 제작자에게 연락하여 사용 방법을 확인하는 것입니다!

safetriggers.py

이 도구를 사용하여 LoRA의 메타데이터에 포함된 TAG, TRIGGER-WORD, TRIGGER-PHRASE를 빠르게 검토하세요.

코드 내에서 "<LoRA 이름>.safetriggers" 텍스트 파일을 읽으려면 다음 코드를 사용하세요:

import os
import json

with open('lora.safetriggers') as f: 
    data = f.read()

lora_triggers_dictionary = json.loads(data)

print(lora_triggers_dictionary)

면책 조항:

현재 버전은 일부 일반적인 LoRA 형식을 읽고 분석할 수 있지만 모든 형식을 지원하지는 않습니다.

후속 버전을 확인하세요.

참고...

초기 버전은 편두통에서 회복 중일 때 작성된 프루브 오브 콘셉트이며, 일부 코드 결정은 마법의 8공을 사용하여 내려졌습니다. 지금은 그 결정에 후회하고 있습니다! 더 나은 후속 버전을 사용하세요.