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

Stable Diffusion 설치없이 브라우저에서 동작 가능한「Web Stable Difusion」등장

by 두우우부 2023. 3. 18.
반응형

 

이 프로젝트는 웹 브라우저에 안정적인 확산 모델을 제공합니다. 

모든 것이 서버 지원 없이 브라우저 내에서 실행됩니다(브라우저에서 완전히 실행되는 것은 세계 최초).

 

단, WebSD는 아직 Windows 지원이 되지 않습니다.

 

M1/M2 GPU가 있는 Mac에서만 사용 가능.

Windows는 드라이버 문제로 인해 현재 작동하지 않습니다.

WebGPU가 발전함에 따라 지원이 확대될 것으로 예상. 

 

이미지 생성 AI의 Stable Diffusion을 움직이기 위해서는 충분한 성능의 GPU와 VRAM이 요구되므로, 고사양 PC나 워크스테이션을 사용하거나 GPU 서버에 액세스 하여 연산 리소스를 빌려야 합니다. 엔지니어를 위한 머신러닝 강의를 제공하는 Machine Learning Compilation은 서버 지원 없이 브라우저에서 Stable Diffusion을 실행할 수 있는 Web Stable Diffusion을 공개하고 있습니다.

WebSD | Home
https://mlc.ai/web-stable-diffusion/

 

WebSD | Home

Home Github Web Stable Diffusion This project brings stable diffusion models to web browsers. Everything runs inside the browser with no need of server support. To our knowledge, this is the the world’s first stable diffusion completely running on the br

mlc.ai


Web Stable Diffusion은 데모 버전이 공개되어 있지만, M1 또는 M2 탑재의 Mac에서만 동작이 확인되고 있습니다.

 

이번에는 M1 탑재 iMac(8 코어 CPU · 8 코어 GPU · 256GB 스토리지 · RAM 16GB 모델)에서 실제로 데모를 돌려 보았습니다.

Web Stable Diffusion의 데모 버전은 Google Chrome의 시험 운용판인 Chrome Canary에서 동작한다고 하여, Chrome Canary의 배포 페이지에 액세스해, 「Chrome Canary 」를 다운로드하여 설치하겠습니다.


동의 및 설치를 클릭하여 DMG 형식의 설치 프로그램을 다운로드합니다. 설치 프로그램의 파일 크기는 218.6MB입니다.


설치 프로그램을 시작하여 Google Canary를 설치합니다.

 

그런 다음 터미널에 "/Applications/Google\Chrome\Canary.app/Contents/MacOS/Google\Chrome\Canary --enable-dawn-features=disable_robustness"를 입력하여 Chrome Canary를 시작합니다.


Web Stable Diffusion 페이지로 이동합니다.


페이지 하단의 "Demo"에 Web Stable Diffusion 데모가 설치되어 있습니다. Input prompt에 프롬프트를 입력하고 Negative prompt (optional)에 네거티브 프롬프트를 입력합니다. 

 

이번에는 기본 "A photo of an astronaut riding a horse on mars (화성에서 말을 타는 우주 비행사의 사진)" 프롬프트 그대로 "Generate"를 클릭합니다.

 

이미지 생성이 시작되었습니다.

대체로 5분 미만으로 다음과 같은 이미지가 생성되었습니다.


이번에는 "Select scheduler"에서 스케줄러의 스텝 수를, "Render intermediate steps"로 렌더링의 중간 스텝을 변경했습니다. 스케줄러의 스텝수는 20 스텝과 50 스텝에서 선택할 수 있습니다만, 50 스텝수가 이미지의 정밀도가 올라가는 만큼, 생성 시간이 더 걸립니다. 또한 렌더링의 중간 단계를 설정하면 생성 중인 노이즈가 표시되지만 생성하는데 시간이 오래 걸립니다.

생성이 시작되고 U-Net 인코딩 단계로 가면 노이즈가 표시되었습니다.


조금씩 노이즈에서 이미지로 생성됩니다.


최종적으로 생성된 이미지는 다음과 같습니다. 「389 secs elapsed(389초 경과)」라고 하는 것으로, 이미지를 1장 생성하는데 약 6분 30초 걸렸습니다.


또한 Web Stable Diffusion의 코드는 GitHub에서 공개되어 있습니다.

GitHub - mlc-ai/web-stable-diffusion: Bringing stable diffusion models to web browsers. Everything runs inside the browser with no server support.
https://github.com/mlc-ai/web-stable-diffusion

 

GitHub - mlc-ai/web-stable-diffusion: Bringing stable diffusion models to web browsers. Everything runs inside the browser with

Bringing stable diffusion models to web browsers. Everything runs inside the browser with no server support. - GitHub - mlc-ai/web-stable-diffusion: Bringing stable diffusion models to web browser...

github.com

반응형