FluxGym은 간단하고 직관적인 웹 인터페이스를 통해 자신의 컴퓨터에서 Flux 용 LoRA를 직접 학습시킬 수 있는 오픈소스 도구입니다.
설치하는 방법은 여러가지가 있지만, Stability Matrix 를 사용하면 소프트웨어 구성을 자동적으로 처리해주므로 편리합니다. 이 두가지 소프트웨어를 사용하면 클릭 몇번 만으로 Flux 용 LoRA를 자신의 PC에서 학습시킬 수 있습니다.
 |
 |
| Vintage Pulp Style LoRA | Natalie Portman LoRA |
| 시스템 요구사항 FluxGym은 오픈소스 소프트웨어이지만, 자원이 많이 필요한 계산이 필요하므로, 아래의 조건을 만족하는 컴퓨터가 필요합니다. - NVidia GPU - VRAM 12GB (8GB도 가능하지만 매우 느림) - RAM - 32GB - 윈도 10 또는 11 (64비트) |
FluxGym 설치방법
1 단계: Stability Matrix 설치
Stability Matrix는 Automatic1111, Forge, Fooocus, ComfyUI 등 다양한 Stable Diffusion용 웹 UI 들을 통합적으로 관리하고, 모델을 공유할 수 있는 있는 도구입니다. Stability Matrix가 FluxGym도 지원하므로 이 도구부터 설치합니다.
Stability Matrix 설치 방법은 이 글을 참고하세요.
2 단계: FluxGym 설치
다음으로 Stability Matrix의 좌측 메뉴에서 정육면체 아이콘(Packages)를 누르고 아래쪽에 있는 "+ Add Package" 버튼을 클릭합니다.
다음으로 맨 아래에 있는 [+ Add Package] 단추를 누르면 아래와 같이 나오는데, Training과 FluxGym을 차례로 누르고....
그 다음 화면에서 [Install] 버튼을 누르면 설치가 됩니다. Stability Matrix는 FluxGym 뿐만 아니라, LoRA를 만드는데 필요한 템플릿도 함께 다운로드 받습니다. 이 과정은 몇 분 정도 걸릴 수 있습니다.
FluxGym 실행
설치가 완료된 후, FluxGym의 "Launch" 버튼을 누르면 실행됩니다.
FluxGym을 사용한 LoRA 학습
0 단계: 이미지 준비
생성하고자 하는 LoRA의 개념에 맞는 이미지를 10장에서 30장 정도 수집합니다. 이미지 크기와 품질이 좋을 수록 좋습니다. 이미지는 JPEG과 PNG 포맷이어야 하며, 종류가 다양할 수록 좋습니다.
이미지 준비에 관한 자세한 사항은 이 글을 참고하세요.
1 단계: 학습 설정
FluxGym에서 아래와 같이 설정합니다.
- 모델의 이름: 적당한 이름을 입력합니다. (예: SpongeBob SquarePants for Flux Dev)
- 트리거 단어: 해당 LoRA를 사용할 때 지정하는 단어를 입력합니다. (예: SpongeBobCharacter)
- 사용가능한 VRAM : 자신의 GPU에 맞도록 설정합니다.
- 처음 시도할 때는 다음과 같은 설정을 사용합니다. 참고로 이미지가 많을 경우에는 Repeats를 줄여도 되고, 이미지 수가 작을 경우에는 Repeats를 늘려야 합니다.
- Repeat trains per image(이미지당 학습횟수): 4
- Max Train Epochs(최대 학습 에포크): 16
2 단계: 이미지 추가
화면 가운데를 클릭하고, 이미지를 선택하거나, 직접 Drag&Drop합니다. 이때 이미지와 함께 파일별로 설명(description)이 있다면 함께 선택합니다.
3. 단계: 설명 작성
2 단계에서 txt 파일을 함께 올렸을 경우, 해당 파일속의 설명이 표시됩니다. 아닐 경우, 각각의 이미지에 대한 설명을 직접 입력할 수도 있습니다. 또한 맨 위에 있는 [Add AI captions with Florence-2] 버튼을 누르면 AI가 감지한 이미지 설명이 추가됩니다.
참고로, 이 단계는 건너뛰셔도 됩니다. 실제로 설명 없이도 LoRA를 학습시킬 수도 있으며, 어떤 사용자들은 설명 없이 학습시키는 것이 더 효과적이라고도 합니다.
4 단계: 학습 시작
이제 Start training 버튼을 누르면 학습이 시작됩니다. 학습 속도는 PC 의 하드웨어 사양에 따라 달라집니다. 참고로 저는 RTX 3070 8GB 를 사용하고 있는데, 실행시켜보니 메모리가 부족하다고 나오네요. ㅠㅠ
LoRA 사용방법
학습이 완료되면, 아래의 폴더에 LoRA가 생성됩니다.
StabilityMatrix\Data\Packages\FluxGym\outputs
이 파일을 아래의 LoRA 모델 폴더에 옮겨주면 ComfyUI나 Forge 등에서 사용할 수 있습니다.
StabilityMatrix\Data\Models\Lora
라이선스
FLUX.1 dev를 사용하여 학습시켰기 때문에, 이렇게 생성한 LoRA는 비-상업적 라이선스를 따르게 됩니다.
마무리
FluxGym을 사용하면 간단하면서도 무료로 자기 자신만의 Flux용 LoRA를 생성할 수 있습니다.
상대적으로 고성능의 하드웨어가 필요하기는 하지만, 유연하게 학습시킬 수 있으며, 생성한 모델을 완벽하게 제어할 수 있습니다.
이 도구를 사용하면 Flux의 창의적인 가능성을 확장한, 새로운 LoRA를 만들 수 있습니다. 한계는 오로지 상상력 뿐입니다!
===
이상입니다.
이 글은 stablediffuson.blog의 글을 번역하면서 필요한대로 수정하여 작성한 글입니다.