본문 바로가기
AI · 인공지능/이미지 생성 AI

Intel제 CPU에서도「스테이블 디퓨전(Stable Diffusion)」을 사용 가능한「stable_diffusion.openvino」가 등장

by 두우우부 2022. 9. 4.
반응형

입력한 텍스트에 맞추어 진짜 사진과 같은 고퀄 이미지를 생성해주는 AI가 「Stable Diffusion」입니다. 그런 Stable Diffusion은 로컬 환경에서 이용하려면 NVIDIA제 GPU를 탑재한 단말이 필요합니다. 그런 Stable Diffusion을 Intel제 CPU에서도 실행할 수 있도록 해주는 「stable_diffusion.openvino」가 등장했습니다.

GitHub - bes-dev/stable_diffusion.openvino
https://github.com/bes-dev/stable_diffusion.openvino

GitHub - bes-dev/stable_diffusion.openvino

Contribute to bes-dev/stable_diffusion.openvino development by creating an account on GitHub.

github.com

Stable Diffusion은 웹상에서 사용할 수 있는 데모판 외에 로컬 환경에서 실행하는 것도 가능합니다.


「stable_diffusion.openvino」는 OpenVINO에 대응한 Intel CPU를 탑재한 머신에서 구동이 가능합니다.

OS는 Windows / macOS / Linux에 대응하고 있어, 버전 3.8 이후의 Python 실행 환경만 준비되면 이미지 생성을 실행할 수 있습니다.

아래의 이미지는 「Street-art painting of Emilia Clarke in style of Banksy, photorealism」

「뱅크시(영국의 가명 미술가 겸 그래피티 아티스트)풍으로 그린 에밀리아 클라크(영화 미비포유 여배우)를 스트리트 아트 슈퍼 리얼리즘으로」라고 하는 텍스트로부터, Stable Diffusion이 생성한 이미지입니다.


엔지니어 karaage0703 이 Windows에 WSL2를 설치하고 Windows에서 네이티브 Linux가 실행되는 환경을 구축하고 pyenv 및 pyenv-virtualenv를 사용하여 Python 환경을 구축하여 "stable_diffusion.openvino"를 이용하는 절차를 게시하고 있습니다. Stable Diffusion

위에서 설정 부분을 가져와 봤습니다.

Python 환경은 pyenv/pyenv-virtualenv를 사용합니다.

다음 명령을 실행

$ pyenv global 3.8.12
$ pyenv virtualenv 3.8.12 sd
$ pyenv global 3.8.12/envs/sd 

소프트 설정

git clone https://github.com/bes-dev/stable_diffusion.openvino
cd stable_diffusion.openvino/
pip install -r requirements.txt

사용법

다음 명령으로 이미지를 생성할 수 있습니다.

$ python stable_diffusion.py --prompt "karaage"
ftfy or spacy is not installed using BERT BasicTokenizer instead of ftfy.
32it [01:41,  3.18s/it]

 

키워드 karaage(카라아게)로 생성된 이미지

반응형