Diffusers, JAX/Flax 지원
🧨 Stable Diffusion in JAX / Flax !
·2022.10.13 09:00
Hugging Face Diffusers 라이브러리가 Flax를 지원하여 TPU 환경에서 Stable Diffusion의 빠른 추론이 가능해졌다.
Hugging Face의 Diffusers 라이브러리가 버전 0.5.1부터 Flax를 지원한다. 이를 통해 Google TPU(Colab, Kaggle, GCP 등) 환경에서 Stable Diffusion의 매우 빠른 추론(Inference)이 가능하다.
JAX는 TPU 서버의 8개 가속기를 병렬로 활용할 수 있어 하드웨어 성능을 극대화하는 데 유리하다.
주요 특징은 다음과 같다:
- FlaxStableDiffusionPipeline을 통한 모델 로드 가능
- TPU 최적화를 위한 bfloat16 데이터 타입 지원
- Hugging Face Hub를 통한 모델 가중치 제공 및 CreativeML OpenRAIL-M 라이선스 준수
이 요약은 원문 이해를 돕기 위한 큐레이션입니다. 저작권은 원저작자에게 있으며, 정확한 내용과 맥락은 원문을 확인하세요.
요약 오류, 출처 표기 문제, 삭제 요청은 문의 · 건의로 알려주세요.