본문 바로가기
야매 개발실/StableDiffusion

Stable-Duffison-webui 간단 설치 방법

by 야매플머 2023. 10. 20.
반응형

안녕하세요. 이상한 프로그래머, 야매 프로그래머 입니다.

오늘은 이미지 생성 형 AI, Stable Diffusion 을 자신의 데스크탑 에서 사용 할 수 있도록 설치 하는 방법을 보도록 하겠습니다. 다소 뒷북 일 수 있습니다.

제가 일하는 산업 분야 에서는 이전부터 AI에 관한 관심이 상당히 높은 관심을 보이고 있었습니다. 그리고 올해 들어, 아주 급격한 발전을 보이고 있는 각종 생성 형 AI로 인해 동료 분들 과도 많은 이야기를 나누었습니다.

대부분의 의견은, 당장에는 사회나 산업 전반에 유래가 없었고 가이드 라인이 없다 보니, 여러가지 법적인 문제와 윤리적인 문제들이 발생 할 것 이지만, 결국 우리는 이 AI 도구를 적극 사용 할 수 밖에 없을 것이라는 결론 이였죠.

그래서 저도 이런 AI 도구들 사용에 적응을 하기 위해서, 대표적인 AI 도구들인 ChatGPT, Github Copilot, Stable Diffusion 등 여러가지를 사용해보고 있습니다. 오늘은 간단하게 Stable Diffusion 을 개인 데스크탑 에서 설치하여 사용 할 수 있는 Stable-Diffusion-webui를 설치 하는 방법을 간단 하게 소개 해보려고 합니다.

Stable Diffusion 의 기술적인 내용에 대해 알고 싶어서 이런 저런 정보를 찾아 봤는데, 저는 이 알고리즘 내용을 보고 이해하지 못했습니다. 대략적으로 확률통계를 통해, 디노이징을 거쳐 이미지를 생성한다 정도만 이해 했습니다. 이 도구에서 사용 하고 있는 Diffusion Model 알고리즘은 상당히 이슈가 되고 있는 알고리즘 중 하나 입니다. 관심이 있고 역량이 되시는 분들이라면 충분히 연구 해볼 만한 분야가 아닌가 생각 됩니다.

실제로 이 알고리즘을 기반으로, text-to-image 뿐만 아니라, 대화형 도구, 작곡, TTS, 사운드 생성 등 가능성이 상당 한 것으로 보고 있습니다.

그럼 설치 해보도록 하겠습니다. 사실 Stable Diffsuion 이 공개 되어있는 공식 Github를 방문해 보면, 일반인인 우리 같은 사람이 쉽게 사용 할 수 있도록 인터페이스 구성이 되어 있지 않습니다. 코드 더미로 된 프로젝트 이지요.

<https://github.com/Stability-AI/stablediffusion>

자신만의 커스텀 된 UI를 구성해서 사용해 보고 싶거나, 나는 터미널 콘솔 화면에서 작업 하는 것이 편하신 분들은 여기 공식 코드를 받아서 사용 하시는 것도 가능 합니다.

오늘 여기 에서는 AUTOMATIC1111 님이 만들어 공유해 주신 프로젝트를 이용하여 간단하게 설치하고 사용 하도록 하겠습니다. 일반적인 사용에서는 부족함이 없을 것이기 때문에 함께 설치 해보도록 하죠.

<https://github.com/AUTOMATIC1111/stable-diffusion-webui>

우선 시스템 사양을 간단하게 안내 해 드리자면, VRAM 4GB 이상의 Nvidia 그래픽 카드를 추천 한다고 나와 있습니다. 하지만 제가 사용 해본 결과 VRAM 4GB 는 말 그대로 최소 조건이고, 실제로 사용하기에는 무리가 있습니다. 자신의 데스크탑 에서 사용 하시려면 최소 VRAM 12GB 이상의 Nvidia 그래픽 카드를 사용 하시는 것을 강력 권장 드립니다.

12GB 이상 되어야, 어느 정도 원활하게 결과물을 출력 할 수 있고, 추가 학습도 가능 했습니다. 그리고 무조건 Nvidia 그래픽 카드를 사용 하실 것을 추천 드립니다. 기타 GPU를 이용 해서도 사용이 가능 이라고 기록 되어 있지만 편하게 이용 하시려면 추천 되지 않습니다.

추가로 넉넉한 디스크 용량을 확보 해 두 시길 바랍니다. 도구 자체는 용량이 큰 편은 아니지만 실제 설치를 해보면 의존성 있는 패키지를 설치하고 거기에 더해, 그림 생산을 위한 모델들을 다운로드 받아 보면, 몇 GB 는 쉽게 넘어 가기 때문에 넉넉한 디스크 용량이 필요 합니다.

이제 AUTOMATIC1111 님이 작성해둔 설치 방법을 따라가서 함께 설치 해보도록 하겠습니다. Automatic Installation on Windows 항목대로 설치를 하겠습니다. 이름 그대로 자동 설치이기 때문에 아주 손쉽게 설치 하루 수 있습니다. 우선 Python 3.10.6 버전을 설치 합니다. 다른 파이썬 버전으로 작동 되는지 궁금해서 제가 시도를 해 봤는데요. 그냥 사용 하라고 안내 되어 있는 버전을 사용 하면 됩니다. 그리고 소스 코드를 받기 위한 git 을 설치 하도록 합니다. 두 프로그램 모두 여기 제공 되고 있는 페이지의 링크를 따라 들어가시면 쉽게 다운로드 받아서 설치 할 수 있습니다.

그리고 터미널을 열고 설치를 할 위치로 이동 한 후:

git clone <https://github.com/AUTOMATIC1111/stable-diffusion-webui.git> <.|원하는 디렉토리>

git을 이용하여 코드를 내려 받습니다. git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git <원하는 위치 또는 현재 위치> 괄호 부분에는 원하는 위치 경로를 넣거나, 현재 위치를 원한다면 . 을 넣어주세요.

내려 받기가 다 되었다면 이제 설치가 다 된 것입니다. 이제 내려 받은 디렉토리 로 이동 해서 webui-user.bat를 실행 하면 필요한 패키지들을 자동으로 설치하면서 실행 까지 진행 되게 됩니다.

실행 이후에 바로 사용해보기 위해 실행 전에 학습 모델을 다운로드 받도록 하겠습니다. 이 학습 모델이 없다면 그림을 그리지 못하니까요.

SD1.5 (v1-5-pruned.safetensors) - <https://huggingface.co/runwayml/stable-diffusion-v1-5/tree/main>
SDXL1.0 (sd_xl_base_1.0.safetensors) - <https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/tree/main>

저는 우선 가장 최신 학습 모델인 SD-XL-1.0 (sd_xl_base_1.0.safetensors) 모델과 SD-1.5 (v1-5-pruned.safetensors) 모델을 다운로드 받았습니다. 다운로드 받은 모델 파일은 내려 받은 코드 디렉토리의 models/stable-diffusion 경로에 넣으면 됩니다.

그리고 webui-user.bat 파일을 메모장으로 열어서 set PYTHON= 항목에 python이 설치된 경로를 추가 하겠습니다. 제가 별도로 이 파일에 python 경로를 추가 하는 이유는 여러 버전의 python을 사용하고 있어서 다른 버전들도 시스템에 설치 되어 있기 때문에, stable diffusion 에서 사용할 python 의 경로를 지정해 주는 것입니다. 자 준비가 모두 다 되었네요! 이제 실행 시켜 보겠습니다.

설치가 완료 되었다면, 자동으로 웹 브라우저가 열리면서 stable diffusion 을 사용 할 수 있도록 실행 됩니다. 자동으로 실행 되지 않았다면 실행 중 인 터미널 텍스트 중 접속 할 수 있는 주소가 보일 터이니 해당 주소를 클릭하거나 웹 브라우저 주소 창에 복사해 넣으면 접근 할 수 있습니다. 역시 첫 실행이니 그림을 한번 그려 봐야 갰죠?

네 이렇게 설치가 완료 되었고 작동까지 확인 해 봤습니다. 아주 간단하죠? 여러분들도 설치해서 재미있게 그림을 만들어 보세요. 나중에 추가로 stable diffusion 사용에 관한 팁이나 정보들도 공유할 예정이니까요. 채널에 관심이 생기신다면 구독 하시고 자주 놀러 오세요.

반응형