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)
  • 블로그 메뉴

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

  • 공지사항

  • 인기 글

  • 태그

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

  • 최근 글

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

티스토리툴바