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

2024. 11. 11. 11:48·Etc
728x90
반응형

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

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX. - huggingface/diffusers

github.com

 

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

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  로 각각 분해된 것을 확인할 수 있습니다.

 

 

 

 

728x90
반응형
저작자표시 (새창열림)

'Etc' 카테고리의 다른 글

[lmdb] lmdb 파일 쓰기 및 읽기  (1) 2025.01.23
sd-x2-latent-upscaler 모델로 image upscale 하기  (2) 2024.11.21
[JNI] jbyte를 C++의 vector<uchar>로 변환하는 방법  (0) 2023.03.12
[JNI] JNI 사용법 및 튜토리얼 (2)  (0) 2023.03.12
[JNI] JNI 사용법 및 튜토리얼 (1)  (3) 2023.03.11
'Etc' 카테고리의 다른 글
  • [lmdb] lmdb 파일 쓰기 및 읽기
  • sd-x2-latent-upscaler 모델로 image upscale 하기
  • [JNI] jbyte를 C++의 vector<uchar>로 변환하는 방법
  • [JNI] JNI 사용법 및 튜토리얼 (2)
ga.0_0.ga
ga.0_0.ga
    반응형
    250x250
  • ga.0_0.ga
    ##뚝딱뚝딱 딥러닝##
    ga.0_0.ga
  • 전체
    오늘
    어제
    • 분류 전체보기 (181)
      • Paper Review (51)
        • Video Scene Graph Generation (6)
        • Image Scene Graph Generation (18)
        • Graph Model (5)
        • Key Information Extraction (4)
        • Fake Detection (2)
        • Text to Image (1)
        • Diffusion Personalization (4)
        • etc (11)
      • AI Research (49)
        • Deep Learning (30)
        • Artificial Intelligence (15)
        • Data Analysis (4)
      • Pytorch (10)
      • ONNX (5)
      • OpenCV (2)
      • Error Note (34)
      • Linux (2)
      • Docker (3)
      • Etc (7)
      • My Study (16)
        • Algorithm (10)
        • Project (4)
        • Interview (2)
  • 블로그 메뉴

    • 홈
    • 태그
    • 방명록
  • 링크

  • 공지사항

  • 인기 글

  • 태그

    fine tuning
    Activation Function
    3dinput
    나이브 베이즈 분류
    permute
    알고리즘
    차원의 저주
    dataloader
    오차 역전파
    GCN
    transformer
    contiguous
    RuntimeError
    HRNet
    정규화
    ONNX
    Logistic regression
    i3d
    linear regression
    TypeError
    Inductive bias
    활성화 함수
    JNI
    pytorch
    torch.nn
    dataset
    그래프신경망
    forch.nn.functional
    tensorflow
    pandas
  • 최근 댓글

  • 최근 글

  • hELLO· Designed By정상우.v4.10.0
ga.0_0.ga
safetensor 모델을 diffusers에서 사용 가능하게 변경하기
상단으로

티스토리툴바