이미지 생성 AI "Stable Diffusion"을 Intel Arc에서 구동하는 데 성공

최근, 문장(프롬프트)을 입력하는 것만으로 화상을 생성해 주는 AI「Stable Diffusion」이 큰 주목을 끌고 있습니다만, Stable diffusion은 기본적으로 NVIDIA제의 GPU로 계산을 실시하도록 설계되어 따라서 타사의 GPU로 계산을 실시하는 것은 상정되어 있지 않습니다. 그러나 프로그래머 rahulunair 씨가 개발한「stable_diffusion_arc」로, Intel제 GPU 「Intel Arc」에서 Stable Diffusion을 동작시키는 것에 성공하였습니다.

GitHub - rahulunair/stable_diffusion_arc: Stable Difussion inference on Intel Arc dGPUs

GitHub - rahulunair/stable_diffusion_arc: Stable Difussion inference on Intel Arc dGPUs

Stability AI가 공개하고 있는「Stable Diffusion」은, 입력한 문장(프롬프트)에 따른 화상을 생성할 수 있는 AI입니다.

Stable Diffusion을 사용하려면 NVIDIA의 GPU를 탑재한 컴퓨터를 준비할 필요가 있었기 때문에, 처음에는 그 이용 장벽이 높았고, 누구나 사용할 수 있는 것은 아니었습니다. 그러나 Stable Diffusion은 오픈 소스 개발이 진행되어 비교적 낮은 스펙의 컴퓨터에서도 동작하거나 CPU만으로도 동작, 스마트폰 앱으로 동작하는 등 여러 버전으로 확장되고 있습니다.

이번에 rahulunair는 Intel Arc A770의 16GB 모델을 사용하여 Stable Diffusion을 작동시키고 있습니다.

Intel Arc에서 Stable Diffusion을 실행하려면 기계 학습 라이브러리인 PyTorch 또는 TensorFlow 가 필요합니다. Intel Arc에서 PyTorch를 사용하려면 Intel extensions for PyTorch 또는 ipex의 최신 버전이 필요합니다.

처음 이미지를 생성할 때는 약 60초, 그 이후에는 약 30초가 필요합니다.

다음은 Intel Arc에서 PyTorch를 사용하여 Stable Diffusion을 작동시켜 생성한 이미지입니다.

또한 다음 이미지는 TensorFlow를 사용한 이미지입니다.

덧붙여 「Intel Arc의 셋업은 모두 Linux상에서 실시했습니다」라고 rahulunair 씨는 말하고 있습니다.