Docker로 간편하게 Stable Diffusion XL 셋업하기

May 7, 2024 • 3 min read

블로그의 아이덴티티를 추가하기 위해서는 나만의 컨텐츠를 추가하는 것이 필요하다고 생각했다. 그리고, 글도 중요하지만, 그림과 같이 시각적인 컨텐츠도 중요하다.

따라서, 나만의 그림 컨텐츠를 만들기 위해서 인공지능을 사용하기로 했다. 누구나 손 쉽게 따라 할 수 있도록 docker 를 사용해서 stable diffusion 모델을 셋업하는 방법을 정리했다.

Github 에서 레포 설치

위 깃허브 링크에 있는 레포를 사용해서 도커를 셋업했다. GPU가 있는 서버에서 설치를 원하는 폴더로 이동한 다음에, 레포를 다운받자.

git clone https://github.com/AbdBarho/stable-diffusion-webui-docker.git
cd stable-diffusion-webui-docker/

금방 다운이 된다. 레포가 다운된 폴더로 이동하자.

docker compose --profile download up --build

도커 이미지에 있는 파일을 빌드하자. 필요한 데이터가 많아 다운에 시간이 조금 걸릴 수 있다.

docker compose --profile auto up --build

빌드가 끝났다면, 서비스를 실행해보자. 여러가지 옵션이 있는데, 가장 무난한 AUTOMATIC1111 UI를 사용하는 옵션을 사용했다.

서비스가 실행되면, 자동으로 웹브라우저가 실행된다. 만약 웹브라우저가 켜지지 않는다면, 직접 크롬과 같은 웹 브라우저를 실행하고, 주소창에 http://0.0.0.0:7860 를 입력하면 된다.

기본 모델도 어느정도 성능이 나오기는 하지만, 기왕 설치한거 좋은 모델을 사용해보자.

Hugging Face 홈페이지에서 모델 파일을 다운받을 수 있다. 용량이 커서 조금 시간이 걸린다.

stable-diffusion-webui-docker/data/models/Stable-diffusion

다운받은 모델을 위와 같이 설치된 경로를 기준으로 /data/models/Stable-diffusion 경로에 넣어주면 끝난다.