[트러블 슈팅] from diffusers import FluxKontextPipeline 불가 - 100-hours-a-week/5-yeosa-wiki GitHub Wiki

1. 문제

a. FluxKontextPipeline 사용 시 ImportError 발생

  • 오류 메시지 예시:

    ImportError: cannot import name 'FluxKontextPipeline' from 'diffusers'
    

    또는

    Could not import module 'CLIPImageProcessor'. Are this object's requirements defined correctly?
    

2. 원인

  • PyPI로 설치한 diffusers, transformers는 FLUX.1-Kontext 관련 최신 변경사항을 포함하지 않음
  • FluxKontextPipeline은 GitHub의 diffusers 최신 버전에만 존재
  • 또한, transformers 최신 버전 (예: 4.41 이상)이 필요함
  • 추가로, torch==2.7.0sentence-transformers, fastai, torchaudio 등과 호환되지 않아 런타임 충돌 발생 가능

3. 해결 방법

  1. 라이브러리 최신 Git 버전으로 설치

    pip install git+https://github.com/huggingface/diffusers.git
    pip install git+https://github.com/huggingface/transformers.git
    
  2. 의존성 패키지 설치

    pip install accelerate safetensors bitsandbytes
    
  3. torch 버전 명시적으로 재설치 (2.6.0 권장)

    FLUX 및 기타 라이브러리와의 호환성을 위해 아래 명령어로 설치:

    pip install torch==2.6.0 torchvision==0.17.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu118
    
  4. Colab 환경에서 실행 시 주의

    • pip install 순서를 잘 지키지 않으면 충돌 발생
    • 가능하면 가상환경 사용 또는 런타임 초기화 후 재시도