https://github.com/hollowstrawberry/kohya-colab/issues/120


컴맹이지만 대충 읽어보면 jax버전 관련해서 문제가 생긴다는 듯...?


대충 외국인 중 한명이 !pip install jax==0.4.23 jaxlib==0.4.23 -f 를 실행해보라해서 해보는중..


-----------------------------------------------------------------


1. 태거만 쓰는 경우 jax버전만 수정해도 정상작동 (코드를 수정해도 되고 한번 오류난다음에 코드추가해서 다시 돌려도 됨)

!pip install jax==0.4.23 jaxlib==0.4.23


2. 로라 학습을 돌릴경우 jax, xformers 버전 둘다 수정 필요

코드보기 -> xformers 버전을 0.025버전으로 수정, jax 다운그레이드 코드 추가



  !apt -y update -qq

  !apt -y install aria2 -qq

  !pip install torch==2.2.1+cu121 accelerate==0.19.0 transformers==4.30.2 diffusers==0.18.2 bitsandbytes==0.40.0.post4 opencv-python==4.7.0.68

  !pip install pytorch-lightning==1.9.0 voluptuous==0.13.1 toml==0.10.2 ftfy==6.1.1 einops==0.6.0 safetensors

  !pip install huggingface-hub==0.20.3 invisible-watermark==0.2.0 open-clip-torch==2.20.0 dadaptation==3.1 prodigyopt==1.0 lion-pytorch==0.1.2 jax==0.4.23 jaxlib==0.4.23

  !pip install -e .

  if cross_attention == "xformers":

    !pip install -q xformers==0.0.25.dev767


-----------------------------------------------------------------------------------------------------



+수정된 듯..?