스테이블 디퓨전(Stable Diffusion)은 Text-2-Image 기반의 AI 이미지 생성 플랫폼입니다. 이전 포스팅에서 설명한 것처럼 스테이블 디퓨전의 실행 방법은 로컬과 코랩으로 나누어집니다. 로컬 실행은 VRAM 최소 6GB(권장 8GB)의 GPU와 최소 8GB(권장 16GB)의 RAM을 권장합니다. 이 사양에 미치지 못한다면 코랩을 사용합시다.
코랩은 구글의 클라우드 서버의 GPU와 TPU를 활용하는 서비스입니다. 즉 가상의 노트북을 대여한다고 생각하시면 됩니다. 저장 공간은 구글 드라이브 서비스로 확보합니다. 지금부터 코랩을 활용한 스테이블 디퓨전 실행 방법을 알아보겠습니다.
스테이블 디퓨전의 실행 (코랩)
코랩 가입
우선 코랩에 가입합니다. 9.99 달러/월의 요금제를 추천합니다. 무료 이용은 컴퓨터 리소스의 제한이 있습니다. GPU & TPU 사용 시간 및 성능 제한 때문에 갑자기 컴퓨터 세션이 만료될 수 있습니다. (링크)
원클릭 코랩
스테이브를 디퓨전을 쉽게 실행하기 위한 원클릭 코랩 링크를 소개합니다. (링크)
다음 화면에서 설정할 것은 딱 2가지입니다. Ngrok이라는 서비스에 가입하고 API키를 얻습니다. 스테이블 디퓨전의 접속 및 실행 속도를 개선하는 용도입니다. 귀찮다면 건너뛰어도 무방합니다. 다음으로 원클릭 코랩 버튼을 누릅니다. 스테이블 디퓨전이 실행될 때까지 기다립니다.
Ngrok 가입과 API 키 얻기
다음 링크로 접속하여 Ngrok에 가입합니다. 구글이나 GitHub 계정을 이용하여 쉽게 가입할 수 있습니다.
Ngrok에 접속하면 좌측 상단에 Getting Started를 누르고 Your Authtoken을 클릭합니다. 화면 가운데에서 코드를 확인할 수 있습니다. copy 버튼을 눌러서 복사합니다.
스테이블 디퓨전 실행
Ngrok API token 입력창에 복사한 코드를 붙여넣습니다.
스크롤를 올리다 보면 원클릭 코랩이라는 버튼을 확인할 수 있습니다. 실행 버튼을 누릅니다.
잠시 후, 구글 드라이브 연결을 확인하는 팝업 창이 뜨면 연결을 누릅니다.
이제 스테이블 디퓨전 코랩 실행을 위해 기다려야 합니다. 다음 화면이 뜨면, 하단의 URL을 눌러서 스테이블 디퓨전을 실행합니다.
스테이블 디퓨전을 실행한 화면입니다. 스테이블 디퓨전 실행 중에는 코랩 화면을 닫지 않습니다.
실행을 중지하려면 원클릭 코랩 버튼을 클릭하여 멈추고, 화면 상단의 런타임-런타임 연결 해제 및 삭제를 꼭 클릭합니다. 코랩 서버의 GPU 연산 작업을 멈추기 위한 명령이므로 잊지 마시길 바랍니다.
모델의 설치
모델의 종류
모델이라는 것은 스테이블 디퓨전에 세부적인 깊이를 더하는 일종의 플러그인입니다. 간단히 설명하면,
- Checkpoint: 학습된 모델로 실사, 반 실사, 2D 등의 다양한 스타일을 제공합니다.
- Lora: 신체, 얼굴, 자세, 동작 등 대상의 일부분을 학습한 플러그인
- VAE: 빛과 후보정에 관련된 모델
관련 사이트
보통 다음 두 사이트에서 모델을 다운로드 받습니다.
- CivitAI: https://civitai.com
- Hugging Face: https://huggingface.co/models
정리
스테이블 디퓨전의 로컬과 코랩 실행에 관한 정리가 끝났습니다. 다음 포스팅에서는 모델 적용 및 기본적인 이미지 생성 방법을 설명하겠습니다.