AI Image/News & Tips

[Stable Diffusion] AI 이미지 생성을 위한 무료 사이트: Kaggle

PLAI 2024. 8. 24. 16:10

원래 구글 Colab을 썼었지만 아래와 같이 결제한 사람들도 이용을 막는 분위기였다.

https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Online-Services

 

이를 대체해서 사용할 수 있는 사이트로 Kaggle이 있다. 휴대폰 인증을 하면 주 30시간 무료로 GPU 환경을 사용할 수 있다 (용량은 20GB).

https://www.kaggle.com/

 

Kaggle: Your Machine Learning and Data Science Community

Kaggle is the world’s largest data science community with powerful tools and resources to help you achieve your data science goals.

www.kaggle.com

 

Kaggle 가입 이후 코드를 실행하여 Stable Diffusion 모델을 통해 이미지를 생성할 수 있다. 이때 Stable Diffusion Web UI나 ComfyUI를 사용할 수 있는데 (모두 Stable Diffusion 모델을 사용할 수 있는 환경이지만 UI가 다르다고 이해하면 된다), Stable Diffusion Web UI는 dependency 문제인지 에러 때문에 실행이 되지 않았다. 그래서 ComfyUI로 진행했다.

 

https://github.com/comfyanonymous/ComfyUI/blob/master/notebooks/comfyui_colab.ipynb

 

ComfyUI/notebooks/comfyui_colab.ipynb at master · comfyanonymous/ComfyUI

The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface. - comfyanonymous/ComfyUI

github.com

 

위 코드를 Kaggle Notebook에 복사한 뒤, 아래와 같이 Session options에서 ACCELERATOR 'GPU T4 x2'를 선택하고, Internet on을 체크하고 실행하면 된다.

 

코드를 보면 첫번째 셀이 실행 환경을 다운로드하고, 두번째 셀에서 모델 등을 다운로드한다. 세번째 셀부터 어떻게 실행할지 정해서 실제로 실행하는 부분인데, 아래와 같은 선택지가 있다.

1. Run ComfyUI with cloudflared (Recommended Way)
2. Run ComfyUI with localtunnel
3. Run ComfyUI with colab iframe (use only in case the previous way with localtunnel doesn't work)

이 중 1번은 진행이 되지가 않아서 2번으로 진행했다. "Run ComfyUI with localtunnel" 셀을 실행한 후 나타나는 URL을 클릭하면 아래와 같이 ComfyUI를 사용할 수 있는 화면을 확인할 수 있다.

 

주의해야할 점은 Kaggle에서는 성인용 컨탠츠를 만들 경우 정지를 하기 때문에 negative prompt에 NSFW 등의 태그를 넣어줘야 한다.

반응형