safetensor 모델을 diffusers에서 사용 가능하게 변경하기

diffusion을 사용하시는 분들이라면  civitai에서 다양한 모델을 다운받아 사용하실텐데요. 

civitai에서는 모델을 .safetensors의 형태로 제공합니다. 하지만 diffusers에서는 scheduler, text_encoder, tokenizer, unet, vae 가 각각 다른 폴더에 저장되어 있는 파일 구조를 원합니다. 

이를 위해 diffusers에서 제공하는 convert_original_stable_diffusion_to_diffusers.py를 사용하면 됩니다. 코드는 아래 github에서 제공하고 있습니다. 

https://github.com/huggingface/diffusers/blob/main/scripts/convert_original_stable_diffusion_to_diffusers.py

diffusers/scripts/convert_original_stable_diffusion_to_diffusers.py at main · huggingface/diffusers

 

실행 방법은 아래와 같습니다. 

python scripts/convert_original_stable_diffusion_to_diffusers.py \
    --checkpoint_path sd_xl_base_1.0.safetensors \
    --dump_path ./output_dir \
    --pipeline_class_name StableDiffusionXLPipeline \
    --from_safetensors

 

코드 실행 후 sd_xl_base_1.0.safetensors 파일이 output_dir 폴더에 scheduler, text_encoder, tokenizer, unet, vae  로 각각 분해된 것을 확인할 수 있습니다.