stable diffusion SDXL-Turbo 사용하기

https://github.com/Stability-AI/generative-models

GitHub - Stability-AI/generative-models: Generative Models by Stability AI

일단 위 깃허브 repository를 clone해주고, Readme에 있는 설명대로 따라하면 된다.

SDXL-Turbo

1. Follow the installation instructions or update the existing environment with pip install streamlit-keyup.

   - 이부분은 그냥 설명대로 따라하면 된다.
   - Readme 하단 부분에 installation을 통해 필요한 것들을 설치한다.
   - pip install streamlit-keyup도 해준다.

2. Download the weights and place them in the checkpoints/ directory.

- weights 링크로 통해 들어가면 Hugging Face페이지가 나오는데, 오른쪽에 점3개 아이콘을 클릭하고 Clone repository를 통해 파일을 받으면 된다.
- 따로 Python 코드로 다운받는 방법도 있는데, 이는 생략하겠다.

3. Run streamlit run scripts/demo/turbo.py

- 이제 마지막 단계로 turbo.py를 실행시켜준다.
- 그럼 다음 에러를 보게 될 확률이 매우매우 높다. ModuleNotFoundError: No module named '~~~~'
- 다른 아무 설치도 없이 실행한다면 No module 에러를 엄청 많이 볼 수 있으니, 미리 어떤 걸 설치하면 될 지 적어 놓겠다.

pip install omegaconf
pip install clip
pip install hatch
pip install kornia
pip install pytorch_lightning
pip install scipy
pip install torch
##inwatermark 설치
pip install transformers==4.19.2 diffusers invisible-watermark
pip install -e .
## open_clip 설치
pip install open_clip_torch

- 빼먹은 부분이 있을텐데, 그부분은 pip install ~~로 바로 해결 가능할 것이다.

다음 같은 창이 뜬다면, 성공이다!

Turbo 실행 페이지

그리고 cmd창을 보면 다음같이 무언갈 또 설치 중일 것이다. 시간은 20~30분 이상 걸리는 듯 한데, 일단 기다려보자.