본문 바로가기
IT/Python

10분만에 AI 그리기 서비스 만들기(Stable Diffusion 사용법)

by DOSGamer 2022. 9. 22.
반응형

AI 서비스 중에 오픈소스로 공개된 Stable Diffusion 을 Colab 으로 연계해서 서비스를 바로 만들어 돌려볼 수 있다

장점

10분안에 만들 수 있다

로컬환경은 웹브라우저만 띄울 수 있으면 된다

단점

colab 제한으로 하루에 100장 정도 만들면 GPU 제한 걸린다

waifu diffusion 을 사용해서 img2img 가 아직 안된다

(seed 연결로 변화된 이미지 생성이 아직 안됨)

만드는 방법

waifu-diffusion 설명 페이지 접속

colab 으로 연결

사본 복사

라이브러리 설치 및 실행

서비스 도메인 연결 및 사용

waifu-diffusion 설명 페이지 접속

https://huggingface.co/hakurei/waifu-diffusion

위 사이트 접속하면 됩니다

colab 으로 연결

waifu-diffusion 설명내용을 보면

open in Colab 버튼이 보이는 데 누릅니다

사본 복사

waifu-diffusion.ipynb 의 사본을 (구글)드라이브에 저장합니다

그래야 소스 수정해도 본인한테 남아있습니다

라이브러리 설치 및 실행

화살표 클릭하면 실행되면서

라이브러리가 colab 설치됩니다 (colab 런타임 리셋되면 다시 설치해야 합니다)

!pip install transformers gradio scipy ftfy "ipywidgets>=7,<8" datasets diffusers

서비스 도메인 연결 및 사용

하단에 셀 실행합니다

그러면 열심히 관련파일들이 다운로드 되면서 설치됩니다

엄청나게 많이 다운로드 됩니다

설치완료 되면 로컬 URL 과 gradio.app 관련 도메인이 나오는데

OOOOO.gradio.app 로 접속하면 됩니다. (임시도메인임)

https://20270.gradio.app 으로 접속해서

industrial city, by Yoshitaka Amano, akihiko yoshida, akihito tsukushi, John Berkey, by Alan Lee, by Arthur Rackham, by Greg Rutkoswki, sci-fi, dieselpunk, solarpunk

문구를 입력하고 RUN 실행한 결과

항상 1분 이상 정도 걸립니다.

industrial city - > space city 로 변경하고 돌린 결과

colab 리소스 보기에서 GPU 가 열심히 일한 흔적이 보입니다

항상 이미지 생성하면 100% 로 일함


Uploaded by N2T

반응형