스테이블 디퓨전 체크포인트
스테이블 디퓨전에서 원하는 이미지를 만들기 위해서는, 정교한 명령어(프롬프트) 작성뿐만 아니라 다양한 외부 기능을 효과적으로 활용하는 것이 중요합니다.
이번 포스팅에서는 스테이블 디퓨전에서 생성되는 이미지의 스타일과 화풍에 중대한 영향을 미치는 체크포인트에 대해 자세히 살펴본 후, CivitAI에서 체크포인트를 다운받아 스테이블 디퓨전에 적용하는 방법을 설명하겠습니다.
🛎️ 스테이블 디퓨전 체크포인트는 모델(Model)이라는 용어로 불리기도 합니다.
체크포인트란?
체크포인트(Checkpoint)는 개발자들이 AI 모델을 학습하는 과정에서, 중요한 지점마다 모델의 상태(가중치와 파라미터)를 저장한 상태를 의미합니다.
이렇게 저장된 체크포인트는 학습 과정에서 발생할 수 있는 문제(시스템의 문제로 인한 중단 등)에 대비하여 학습의 진행 상태를 보존하며, 추후 학습을 재개하거나, 특정 시점의 모델을 평가하고 사용하는 데 활용됩니다.
스테이블 디퓨전은 오픈 소스로 공개된 AI 이미지 생성 모델입니다. 개발자들은 공개된 모델을 학습시켜 자신만의 모델을 개발하는 과정에서 특정 스타일의 이미지를 생성할 수 있는 상태를 체크포인트로 저장하고 배포할 수 있습니다.
스테이블 디퓨전 체크포인트의 역할
수많은 이미지를 포함한 다양한 데이터를 학습한 체크포인트는 스테이블 디퓨전의 뇌와 같은 역할을 합니다.
인간이 뇌가 없이는 생각하고 행동할 수 없는 것과 마찬가지로, 스테이블 디퓨전도 체크포인트 없이는 이미지를 생성할 수 없습니다.
개인마다 각기 다른 뇌 구조와 경험을 가지고 있는 것처럼, 스테이블 디퓨전 체크포인트도 학습한 데이터의 종류에 따라 생성하는 이미지의 스타일과 화풍에 차이를 보입니다.
스테이블 디퓨전 체크포인트 다운로드
스테이블 디퓨전 체크포인트를 다운로드할 수 있는 가장 대표적인 사이트는 Hugging Face와 CivitAI입니다.
Hugging Face는 자연어처리(NLP), 이미지 생성, 음성 인식 등 다양한 AI 분야의 모델을 제공하는 반면 Civitai는 스테이블 디퓨전에 특화되어 있습니다.
스테이블 디퓨전 입문자에게는 Civitai를 추천드립니다. CivitAI는 UI가 굉장히 직관적이어서 누구나 쉽게 익힐 수 있습니다.
CivitAI 사용법
지금부터 CivitAI에서 체크포인트를 다운받아 스테이블 디퓨전에 적용하는 방법을 설명하겠습니다.
1. CivitAI 접속 → Models 클릭 → Filters 클릭 후 Checkpoint 선택
2. 마음에 드는 체크포인트 선택
왼쪽 위 Animagine XL V3 모델을 선택해 보겠습니다.
Animagine XL V3 체크포인트를 선택하면 보이는 화면입니다. 체크포인트에 대한 기본 정보를 확인할 수 있습니다.
페이지 아래쪽에서는 해당 모델로 생성된 다양한 이미지를 확인할 수 있습니다.
3. 체크포인트 다운로드 버튼 클릭
4. 체크포인트 다운로드 경로 설정 → 다운로드
가장 대표적인 AUTOMATIC1111(WebUI)를 사용하신다는 경우 Stable-diffusion-webui > models > Stable-diffusion 폴더 안에 체크포인트를 저장합니다.
5. 다운로드 완료
무려 6.7GB가 넘는 체크포인트가 다운로드 되었습니다. 기존에 사용 중이던 체크포인트 또한 4.1GB, 5GB로 용량이 매우 큰 편입니다.
6. 스테이블 디퓨전 실행 → 체크포인트 선택
프롬프트 입력창 바로 위에 위치한 체크포인트 옵션 선택 창에서 새로운 체크포인트를 선택합니다.
지금까지 스테이블 디퓨전 체크포인트에 대한 정의와 사용법에 대해 함께 알아보았습니다.
스테이블 디퓨전 체크포인트를 다운로드할 수 있는 CivitAI에는 스테이블 디퓨전으로 생성된 다양한 이미지와 함께 이미지 생성에 사용된 프롬프트가 공유되어 있습니다.
실제로 미드저니, 달리3 유저들도 CivitAI의 이미지를 참조해서 이미지를 생성하기도 합니다. CivitAI에 방문해서 다양한 이미지를 감상하며 영감을 얻어보시기 바랍니다.