【工具】将 SDXL 的 ckpt 文件转为 Diffusers 库格式

最新推荐文章于 2024-06-25 10:20:44 发布

橙子的科研日记

最新推荐文章于 2024-06-25 10:20:44 发布

阅读量858

点赞数 13

文章标签：计算机视觉 python

本文链接：https://blog.csdn.net/bikahuli/article/details/134951200

版权

def convert_sdxl_to_diffusers(pretrained_ckpt_path, output_diffusers_path):
    import os
    os.environ["HF_ENDPOINT"] = "https://hf-mirror.com"  # 设置 HF 镜像源（国内用户使用）
    os.environ["CUDA_VISIBLE_DEVICES"] = "1"  # 设置 GPU 所使用的节点

    import torch
    from diffusers import StableDiffusionXLPipeline
    pipe = StableDiffusionXLPipeline.from_single_file(pretrained_ckpt_path, torch_dtype=torch.float16).to("cuda")
    pipe.save_pretrained(output_diffusers_path, variant="fp16")