AI로 그림 그리기, Google Colab

 

artificial intelligence 키워드만 넣어서 그려낸 그림

 

최근 가장 핫한 이슈로 떠오르는 것은 Chat GPT와 text-to-image가 아닐까 생각된다.

오늘은 그 중에서도 키워드만 입력하면 그림을 그려주는 tti에 대해 얘기해보고자 한다.

 

가장 대표적인 모델인 Stable Diffusion 모델을 사용하는데, 방식은 여러가지가 존재한다.

 

1. Google Colab 사용

2. 기능이 탑재된 웹사이트에서 생성

3. StableDiffusion webui를 설치하여 사용

오늘은 1번 방법인 Google Colab을 이용한 방법으로 진행해보려 한다.

 

2번을 제외한 방법은 Hugging face라는 웹사이트에 회원 가입이 필요하다.

https://huggingface.co/

Hugging Face – The AI community building the future.

 

 

아이디를 만들고 이메일 확인까지 마쳤으면, Access Tokens을 띄워놓고 진행하도록 하자.

중간에 라이센스 허용하는 부분이 나오면 허용 해주면 된다.

 

이제 Colab 페이지로 가보도록 하자.

https://colab.research.google.com/drive/1Iy-xW9t1-OQWhb0hNxueGij8phCyluOh

StableDiffusionUI-Voldemort.ipynb

당연한 얘기겠지만, 구글 로그인은 되어 있어야 한다.

 

 

처음 들어가면 이런 화면이 나오는데 나는 다크모드라서 검은색 배경이다.

그건 무시하고 이제 차례 차례 진행하면 된다.

 

 

 

!nvidia-smi 왼쪽 실행 버튼을 누르고 무시하고 계속하기를 누른다.

 

 

이 처럼 초록색 체크표시가 뜨면 완료된 것이다.

다음으로 넘어가자.

 

 

 

import os 부분을 실행시켜 준다.

주석처리된 부분을 읽어보면 알겠지만, 충돌이 일어나더라도 필요한 수순이고 걱정하지 마라고 한다.

세션이 다운되었다고 뜨더라도 그냥 진행하면 된다.

 

 

!git clone 부분을 실행시켜 준다.

ai모델을 가져오는 작업이라고 보면 된다.

 

 

Normal 1.4 model 부분은 그냥 실행시키면 안되고, 오른쪽에 user_token을 기입해줘야한다. 상단에서 작업 했었던 Hugging Face의 Access Token 부분을 복사해서 붙여넣기 한 다음 실행할 것.

 

 

 

마지막으로 제일 밑에 있는 명령어인 해당 부분을 실행시키면 된다.

 

 

 

이 라인과 착각하지 말 것.

 

 

작업이 완료된 후엔, Running on public URL 부분에 있는 주소를 클릭해 접속하면 된다.

이 링크는 72시간 후에 만료되기 때문에 사용에 제한이 있고, 아무 행동도 하지 않으면 세션 만료도 일어나기 때문에 귀찮은 편이다.

 

이 녀석이 계속해서 도는 것을 볼 수 있는데, Colab을 끄게 되면 ai그림도 그릴 수 없게 되는 점 참고 바란다.

 

 

 

계속해서, AI그림을 시작해보자.

 

 

링크로 접속하면 이런 창을 볼 수 있는데 여기서부터 text to img가 가능해진다.

 

이제 Prompt 창에 키워드를 입력하면 되는데, 영어로만 입력해야 한다.

키워드를 찾는 것이 생각보다 어렵기 때문에 이런 키워드를 찾아나가는 연습과 사이트의 도움이 필요하다.

키워드를 찾으려면 lexica에 방문해보길 바란다.
https://lexica.art/

Lexica

 

Lexica에서 이러한 Prompt를 복사해서 내 Stable Diffusion에 넣는다고 해서 같은 화풍이나 그림이 나오지는 않는다. 왜냐면 학습된 모델이 다르기 때문이다.

Lexica는 자체 모델인 Lexica Aperture v2를 쓰기 때문이다. 하지만 Prompt가 정밀하고, Negative prompt도 정밀하게 넣어준다면 대체적으로 원하는 그림을 얻을 수는 있다.

 

 

같은 키워드로 넣어서 얻은 결과 이미지이다. 느낌은 다르지만 생각보다 괜찮은 이미지를 얻을 수 있다.

이것까지는 단순 맛보기의 느낌인데, 여기서부터 세밀한 조정과 학습을 시킨다면 원하는 화풍과 결과를 얻을 수 있게된다.

 

이후 다음 시리즈부터 모델의 세부 설정 기능, 간단하게 이미지를 얻는 방법, 내 PC에 설치하여 사용하는 방법, 튜닝 방법 등을 소개해보도록 하겠다.