웹툰AI : Stable Diffusion 모델 사용하기

스테이블 디퓨전을 무사히 설치했다면 이제 모델을 불러와 사용하는 방법에 대해서 알아 봅니다. 처음 스테이블 디퓨전을 실행 시키면 기본으로 제공되는 모델인 v1-5-pruned-emaonly가 로드되어 있습니다. 하지만 저는 기본모델 보다는 외부에서 학습한 모델을 불러와 교체해서 사용하는 벙법을 사용하고 싶은 겁니다.

 

학습된 모델 다운로드 받기

딥러닝 학습 된 AI 모델을 공유하는 사이트들이 많이 있습니다만 오늘은 그 중에 civitai.com이라는 곳에서 모델을 다운로드 받을 것입니다. 이 곳 시비타이는 비교적 많은 사용자들이 모델을 공유하고 있는 활발한 사이트 입니다. 이 곳에서 모델을 다운받기 위해서는 가입을 해야 합니다. 가입은 어렵지 않으니 가입 절차를 소개하는 것은 생략 하도록 하겠습니다. 참고로 다소 등급이 높은 AI생성 이미지들도 있으므로 후방주의해야 합니다.(기본적으로는 블라인드 처리 됩니다)

 

Civitai | Stable Diffusion models, embeddings, hypernetworks and more

가입을 마치고 원하는 스타일의 이미지를 클릭해보면 다음과 같이 이미지와 연관된 정보들이 뜹니다.

시비타이 메인화면 이미지 카드를 클릭하면 해당 모델 페이지로 이동 합니다.

이번 과정에서는 MeinaMix 모델을 다운 받아 사용해 볼겁니다. 그래픽 카드가 낮은 경우 메이나 버전을 V7이나 그 이하로 낮춰 사용하면 됩니다. 위 화면에서 다운로드(웹페이지 자동 한글번역 화면을 사용중입니다.) 버튼을 클릭하면 모델을 다운로드 받을 수 있습니다.

저는 메이나v7 버전을 다운로드 받았습니다. 메이나 버전 버튼을 누르면 해당 버전 링크 페이지로 이동 합니다.

다운받은 모델 설치하기

무사히 다운로드를 받았다면 이제 이 파일을 스테이블 디퓨전이 설치 된 폴더의 모델 폴더로 이동 시켜야 합니다. 경로는 자신이 설치한 경로의 stable-diffusion-webui\models\Stable-diffusion 입니다. 

모델이 들어가는 폴더 경로 입니다. 저는 참고로 몇개의 모델을 더 받았습니다.

모델을 무사히 폴더로 이동시켰다면 이제 스테이블 디퓨전을 실행 시킵니다. 이미 실행 중이라면 모델 업데이트를 하면 됩니다. 스테이블 디퓨전을 실행하는 방법은 아래 스테이블 디퓨전 설치 방법을 소개한 페이지를 참조 하세요. 바탕화면에 바로가기 아이콘을 생성해 두면 매우 편합니다.

 

웹툰AI : 내 PC에 Stable Diffusion WebUI 설치하기

스테이블 디퓨전이 실행 중인 웹 페이지에서 모델 선택 칸 옆에 있는 다시읽기 버튼을 클릭합니다. 그럼 잠시 뭔가가 열심히 돌아가다가 완료가 됩니다. 이 과정을 통과하면 이제 모델들을 고를 수 있게 되었습니다.

 

이미지 생성을 위한 프롬프트 복사

이제 AI에게 내릴 명령어(프롬프트)를 입력해야 합니다. 처음이니까... 다른 사람들의 프롬프트를 시험해보는 것도 괞찬을 것 같습니다. 시비타이 사이트에서 프롬프트를 복사하는 벙법을 알아 봅니다.

위 그림처럼 이미지 아랫부분에 흰색으로 ⓘ 표시가 된 곳을 누르면 데이터를 복사 할 수 있는 창이 뜹니다. 이곳에서 복사 버튼을 눌러주면 복사가 됩니다.(참고로 웹페이지 자동번역 기능을 사용중이라 프롬프트가 한글로 번역 되어 있지만 실제 복사는 영문으로 복사 됩니다) 복사가 되었으면 이제 다시 스테이블 디퓨전 페이지로 갑니다.

 

프롬프트 배열하기

위 그림처럼 프롬프트 창에 커서를 놓고 Ctrl V를 누르면 프롬프트가 복사 됩니다. 여기서 바로 생성하기 버튼 누르면 안됩니다. 다음 과정이 있습니다.

위 그림처럼 스타일 자동배열 버튼을 눌러 배열을 해줘야 합니다. 자 이제 준비가 끝났습니다. 생성버튼을 누르기만 하면 됩니다. 일단은 먼저 이미지를 뽑아 봅니다.

 

이미지 생성하기

명령프롬프트 터미널에서도 열심히 뭔가 진행되는 화면이 보입니다.

컴퓨터, 더 자세히는 그래픽 카드의 사양에 따라 이미지가 생성되는 시간이 다릅니다. 저사양 PC에서는 동일한 프롬프트라도 결과를 내지 못하는 경우도 있습니다. 이 경우 에러메시지를 보면 대부분 메모리 부족으로 나옵니다.

 이렇게 이미지 생성이 완료가 되면 시비타이에서 가져 온 프롬프트의 이미지와 같은 이미지가 생성이 됩니다.  같은 모델과 같은 명령어를 썼으니 당연한 일이기는 합니다. 프롬프트를 바꿔가면서 여러가지 출력결과를 보는것도 좋습니다.

 

내가 생성한 이미지 저장소는?

이렇게 생상 된 이미지는 자동으로 내 PC의 특정 공간에 저장이 됩니다. 이제 이 폴더가 어딘지 찾아 보도록 하죠.

위 그림처럼 폴더 아이콘을 누르면 생성 이미지가 저장 된 폴더로 이동 합니다. 생성 날짜별로 폴더를 구축하고 거기에 이미지를 담아 둡니다. 생성 방법에 따라 각각 다른 폴더에 저장 하지만 어찌 되었든 outputs폴더 안에 저장 되므로 outputs 폴더를 열람하면 자신이 생성한 이미지를 열람할 수 있습니다.

생성한 이미지와 사이트의 이미지를 비교한 화면 입니다.

프롬프트를 바꿔가면서 여러 이미지를 생성해 보았습니다.

 

오늘은 AI Stable Diffusion WebUI에서 모델을 불러와 프롬프트를 넣고 이미지를 생성하는 것에 대해서 알아 봤습니다. 모델의 훈련에 따라 다양한 형태의 이미지를 생성 할 수 있다는 것이 신기하네요.