안녕하세요 오늘도 재도전을 한김에 작성을 해봅니다
이미 몇번 망해보고 로컬에 한계를 느껴 이거저거 해보다 잘 안되서
고민하다
코랩으로 했는데... 한번만에 어느정도 잘 나오긴 합니다.
kohya 공식 리파지토리에 있는 코랩 파일은 아니지만 일단 나오긴 합니다.
Google Colaboratory Notebook
Run, share, and edit Python notebooks
colab.research.google.com
사용순서(틀릴수도 있음) - 어째튼 됨
학습에 사용할 사진 미리 구글드라이브에 zip 파일로 업로드 먼저하고 진행하면
1.1 - mount_drive 체크 후 엔터
2.1 - SD1.x model - model_name: Stable-Diffusion-v1-5선택
3.1 - 실행
3.2 - zipfile_url - /content/drive/MyDrive/images.zip 이미지 압축파일 경로만 입력후 실행
4.2.1 - 실행
4.2.2 - general_threshhold 만 0.85로 올린다음에 실행
5.1 - project_name: 원하는 로라 이름 넣어주면됨
pretrained_model_name_or_path: /content/pretrained_model/Stable-Diffusion-v1-5.safetensors
(경로수정 없이 설치 진행했을 경우 같을거임)
output_dir: /content/LoRA/output
output_to_drive 에 체크해주면됨
실행
5.2 - dataset_repeats: 1 (이건 10으로 올려서 해보다가 코랩터졌음...)
activation_word: 쓰고 싶은 단어쓰면 될듯?
실행
5.3 - lr_scheduler: 이거만 linear로 바꾸고 실행
5.4 - num_epochs: 150
vae_batch_size: 1
train_batch_size: 1
실행
뭐 이순서대로 하고 output 받은거로 테스트를 해봄13 장으로 진행했었고 다음껀 하다가 코랩이 터짐....ㅜㅜ
결과 확인
보면 알겠지만 프롬프트나 시드에 따라 좀 들쭉 날쭉 하기도하고 베이스 모델에 따라 기복이 좀 있는듯함
지난번에 사용한 로라까지 합쳐서 응용하면 대충 이렇게 되는덧?
일단 너무 적은양만 학습시키고 쇼파에서 누워있다 찍었던 사진으로 진행해서 그런지
프롬프트 제어를 하지 않으면 쇼파도 내얼굴과 같이 나옴
참고로 학습시킨 사진은 조카소 바디프로필때 사용한 사진과 동일한 사진으로 진행하였음
조카소 바디프로필 결과물과 비교
학습은 했지만 확실히 많이 부족한 느낌이 남
좀 더 알아보고 추가 수정이나 다음글에서 더 자세히 작성해보도록 하겠습니다.
'Stable Diffusion > LoRA' 카테고리의 다른 글
| [LoRA] Stable Diffusion LoRA 만들기~! - 2 (0) | 2023.12.21 |
|---|---|
| [LoRA]Stable Diffusion LoRA 만들어보기~! - 1 (실패) (1) | 2023.12.20 |
