인공지능 이미지 만들기, Stable Diffusion 모델 (feat. Colab)

Stability AI

 

---

요새 핫한 모델로 Stable diffusion 딥러닝 모델이 있습니다. 

 

최근 Stability AI 스타트업 회사가 Open Source로 립러닝 모델을 풀어버린 것이죠.

이 모델은 바로

 

1. 텍스트로 이미지 생성(Text to image generation)

2. 이미지 변환하기(Image modificate)

 

기능을 가지고 있는 매우매우 퀄리티가 좋다고 보이는 파워풀한 모델인데요.

 

실제로 간단한 텍스트만 치고 이 세상에 하나만 존재하는 나만의 이미지를 생성할 수 있게 됩니다.

(...헉 이미지를 만든 내가 저작권을 가지겠죠? 대박입니다)

 

예를 들어서 제가 최근에 만든 이미지로

토끼탈을쓴 다람쥐

 

4k 카메라에 찍힌 이효리씨를 닮은 사진

이렇게 귀여운 토끼틀을 쓴 다람쥐와, 사람형상도 아주 직접 찍은 사진처럼 만들어 낼 수 있었는데요

 

!! 너무 너무 핫한 이 모델 한 번 사용하는 방법을 간단히 알려드리도록 하겠습니다 💨

 

---

준비 단계

 

1. Colab 아이디 준비

https://colab.research.google.com/?hl=ko 

Google Colaboratory

구글에서 쉽게 GPU를 사용할 수 있는 또, 코딩을 할 수 있게 만든 아주 기가막힌 사이트입니다.

 

이를 저희는 이용할 건데요 준비물은 다들 가지고 있으실 구글 아이디 만 있으면 됩니다.

 

2.  Stable Diffusion Colab 사본으로 만들기

https://colab.research.google.com/drive/1kw3egmSn-KgWsikYvOMjJkVDsPLjEMzl

StableDiffusionUI-Voldemort V1.2.ipynb

 

또한 아주 감사하게도 모든 것을 시도해볼 수 있도록 준비를 해놓아 주셔서, 저희는 이용만 하면 되는데요.

 

우선 위의 홈페이지에 들어가서 Drive 복사를 클릭 후 나만의 사본을 만듭니다.

 

3.  이 모델을 이용을 위해서 Hugging Face 가입 및 승인을 눌러야합니다.

 

접속 후, 계정 생성을 합니다

 

https://huggingface.co/

Hugging Face – The AI community building the future.

 

 

 

이 후, New Token -> read 방식으로 토큰을 생성합니다

 

http://huggingface.co/settings/tokens

Hugging Face – The AI community building the future.

 

그러면 저처럼 이 토큰이 생기는데요, 이 토큰은 나중에 쓰이니 그때 복사하셔서 쓰시면 됩니다!

 

이제 다 오셨습니다. 마지막으로 아래의 stable Diffusion 모델 엑세스 동의까지 마무리 하시면 됩니다!!!

 

https://huggingface.co/CompVis/stable-diffusion-v-1-4-original

CompVis/stable-diffusion-v-1-4-original · Hugging Face

 

---

실행 단계

아까 2번째 준비 단계의 Colab 사본 페이지를 만들어 주셨을 텐데요. 이 페이지에 접속합니다.

 

Colab 런타임 유형 변경

 

gpu를 사용하기 위해 런타임 유형을 CPU에서 GPU로 바꿔줍시다

 

이후 차례차례 ctrl enter을 눌러가며 실시를 하면 되는데 아까 준비 단계, 3단계의 우리의 Token기억하시죠

그 토큰을 Normal 1.4model 옆에 있는 user_token : 란에 입력하시고 실행을 하셔야합니다.

(+ 첫번째 코드인 os.kill 은 colab에서 시작시 원래 crash 오류가 나는게 정상이니 그대로 진행하시면 됩니다.)

 

그렇게 마지막까지 차례대로 정상적으로 실행을 하셨나면.

이러한 마지막 실행문을 만나실 수 있으실 텐데요 여기서, Running on Public URL이라 써져있는 저기

gradio app의 URL을 클릭하시면 됩니다. 

 

그 이후에 이렇게 정상적으로 gradio 사이트에 들어오시게 된다면

 

ID : me

PASSWORD : qwerty

 

 

를 치신다면 성공적으로 stable diffusion 모델을 이용할 수 있게 됩니다!!

---

응용 단계

메뉴바에 txt2img, img2img 외에 여러 이미지 변환법도 있으니, 사용을 하시면 되고.

 

저희가 해볼 것은 txt2img 입니다.!!

Photo of the very handsome tree 라 쳐보니

 

 

 

요러한 소나무 같기도한 아주 멋진 나무 사진을 생성을 하게 되었습니다.

 

---

모두 재밌게 사용 해보시고, 나만의 이미지들을 이용하여 멋진 작품을 생성해보시는 것은 어떠실까요  ╰(*°▽°*)╯