Stable Diffusion을 활용한 삼국지 13 신무장 편집-LoRA 학습 (2)

LoRA 학습을 진행하는데에는 다양한 방법이 있으나 저는 kohya_ss의 코드를 활용했습니다.

Kohya_ss의 UI는 Stable Diffusion webUI와는 독립적으로 실행되기에 별도로 설치해주셔야합니다.

 

4) 학습 폴더 정의 및 LoRA 폴더 생성

gui_user. bat파일을 실행하면 브라우저에서 다음과 같은 창이 열리는데 체크된 부분을 완성하고 Prepare Training Data를 눌러줍니다.

instance prompt, class prompt: 적당히, training images: 정리된 학습 데이터 폴더의 디렉토리 입력, destination~: LoRA 파일을 넣을 빈 폴더 경로 입력

 

5) 학습 파라미터 설정

위의 이미지에서 표시된 부분을 수정하되,

Iteration = (Epoch/Train Batch Size) * Repeats * Image Numbers

-> 결과물 수준 하-3000 중-5000 상-10000 임을 고려하여 수치값 설정,

Caption Extension: .txt

LR Scheduler: constant_with_warmup

LR warmup: 5

Network Rank=Network Alpha=8

을 추천합니다.

Advanced Config에서는 위와 같이 체크 표시를 합니다.

 

6) 폴더 경로 지정

4)에서 만들어놓은 폴더의 경로를 지정해줍니다.

"~/img, ~/log, ~/model"의 형태로 있을텐데 각각을 image, logging, output에 연결합니다.

model output name에는 자신이 원하는 모델의 이름을 입력합니다.

 

7) Source model 선택

가지고 있는 모델에서 학습에 사용할 것을 선택합니다. 위의 이미지에서는 webUI의 기본 모델이 적혀있으나 본인이 가진 여러 모델 중에서 코에이 삼국지의 작화와 어울리는 것을 선택합니다.

마치셨다면 Train model을 눌러 학습을 시작합니다.

참고)

프롬프트 창에서 진행사항을 확인할 수 있고 VRAM 8GB, iteration 10000 기준으로 학습에는 4시간 정도 소요되었습니다.