웹툰AI : 화풍을 만들어주는 Dreambooth 설치하기

Dreambooth 익스텐션은 Stable Diffusion WebUI에서 모델의 화풍을 훈련시키는 플러그인 입니다. 오늘은 Dreambooth의 설치 방법을 알아 보도록 하죠.

---

1. Dreambooth 설치 준비 

Stable Diffusion WebUI를 실행하기 위해서는 webui-user.bat 파일을 실행 시켜야 한다는 것을 지난 설치과정에서 알아 보았습니다. webui-user.bat 파일은 실행 명령을 배치해 둔 파일인데요. 오늘은 이 파일을 수정해야 하므로 먼저 메모장이나 텍스트 편집기를 준비해야 합니다. 

GitHub - d8ahazard/sd_dreambooth_extension

메모장을 준비하고 위에 링크한 사이트로 이동합니다.

---

2. Dreambooth 설치과정

설치과정은 이 사이트를 통해 무언가를 다운받아 설치하는 것이 아닙니다.  자 설치를 시작해보죠.

1) Stable Diffusion WebUI를 실행시켜 브라우저에 화면을 띄웁니다.

2) 익스텐션 탭을 엽니다.

SD화면에서 ①번 익스텐션을 눌러 바뀐 화면에서 ②를 누릅니다.  이어 ③ 버튼을 클릭하면 인스톨리스트가 쭈르르륵 나타납니다. 너무 많은경우 ④번 란에 Dreambooth를 입력하여 찾아 설치를 합니다.(이것으로 설치 끝)

 

3)  배치파일을 만들어 줍니다. 

이렇게 되어 있는 배치파일을 메모장 등으로 열어서 아래 내용을 삽입하여 수정을 해줍니다.

set "TORCH_COMMAND=pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116--extra-index-url https://download.pytorch.org/whl/cu116" set "REQS_FILE=.\extensions\sd_dreambooth_extension\requirements.txt" :: Uncomment below to skip trying to install automatically on launch. :: set "DREAMBOOTH_SKIP_INSTALL=True"

 

그리고 저장을 해 줍니다. 만일을 대비해서 원본파일을 webui-user.bat2 등으로 변경하여 별도로 저장해 둡니다.

 

4) 이제 webui-user.bat를 실행시켜 Stable Diffusion WebUI를 재시동 시켜 줍니다.

---

3. Dreambooth 시동

Stable Diffusion WebUI을 완전히 껐다가 다시 webui-user.bat를 실행해 줘야 정상 작동 합니다.  그럼  스테이블 브라우저 화면에 이전에 없었던 Dreambooth 탭이 하나 생깁니다. 오늘은 여기까지 Dreambooth 익스텐션의 설치법을 알아보았습니다.