스테이블 디퓨전 사용 시 PC 그래픽 카드 및 CPU 등의 하드웨어 사양이 좋지 못해 이미지 생성에 오래 걸릴 경우 다음과 같이 xformers 설정을 하면 보다 더 빠르게 이미지를 생성하실 수 있습니다.
먼저 그래픽 카드 사양에 따른 설정 법은 다음 글을 통해 확인해 보시기 바랍니다. PC에 장착되어 있는 그래픽 카드의 성능을 파악하고 그에 맞는 스테이블 디퓨전 설정을 해야 이미지 생성 작업을 원활하게 하실 수 있습니다.
xformers란?
xformers는 PyTorch와 호환되는 고성능 딥러닝 모델 가속화 라이브러리입니다. 스테이블 디퓨전(Stable Diffusion)과 같은 텍스트-투-이미지 모델에서 이미지 생성 속도를 크게 개선하기 위해 주로 사용됩니다.
xformers를 사용하면 모델 추론 시간 단축, 메모리 사용량 감소 등을 통해 이미지 생성 속도를 대폭 낮출 수 있습니다.
허깅 페이스에 xformers에 대한 자세한 설치 방법이 나와 있으니 공식 문서를 참고하셔도 좋습니다.
xformers 사용 방법
1. webui-user.bat 파일 수정
스테이블 디퓨전 설치 경로에 있는 webui-user.bat 파일의 내용을 다음과 같이 수정합니다.
@echo off
set PYTHON=
set GIT=
set VENV_DIR=
set COMMANDLINE_ARGS=--xformers
call webui.bat
2. 저장 후 닫기
작성 완료되면 저장하고 해당 창을 닫습니다.
3. webui-user.bat 실행
스테이블 디퓨전을 실행할 때 webui-user.bat 파일로 실행합니다.
xformers 실행 오류 해결 방법
webui-user.bat 실행 시 xformers 설치 오류로 인해 실행이 안된다면 다음과 같이 해결해 보시기 바랍니다.
1. webui-user.bat 파일 수정
위에서 webui-user.bat 파일을 수정합니다.
@echo off
set PYTHON=
set GIT=
set VENV_DIR=
set COMMANDLINE_ARGS=--xformers --reinstall --xformers
call webui.bat
2. webui-user.bat 파일 저장 후 닫기
수정이 완료되면 저장 후 닫습니다.
3. webui-user.bat 파일 실행
webui-user.bat 파일을 실행하여 정상적으로 스테이블 디퓨전이 실행되는지 확인합니다.
마치며
xformers는 스테이블 디퓨전과 같은 고성능 딥러닝 모델의 이미지 생성 속도를 크게 향상시켜주는 도구입니다. 이를 사용하여 이미지 생성 작업을 더욱 빠르게 수행할 수 있으니, 위의 방법을 따라 쉽게 적용해보세요.