Windows: 如何实现CLIPTokenizer.from_pretrained`本地加载`stable-diffusion-2-1-base`

参考:https://blog.csdn.net/qq_38423499/article/details/137158458
https://github.com/VinAIResearch/Anti-DreamBooth?tab=readme-ov-file

联网下载没有问题:

python 复制代码
import os

os.environ["HF_ENDPOINT"] = "https://hf-mirror.com"
import torch
from diffusers import DiffusionPipeline

pipe = DiffusionPipeline.from_pretrained(
    "stabilityai/stable-diffusion-2-1-base", torch_dtype=torch.float16
)

从 Hugging Face 上下载指定仓库的模型后,会保存到了C:\Users\***\.cache\huggingface\diffusers\models--stabilityai--stable-diffusion-2-1-base文件夹下面,总共14个文件

但是,我们进行CLIPTokenizer.from_pretrained进行本地路径加载stable-diffusion-2-1-base时,报错:

起初,按照要求将stable-diffusion-2-1-base稳定扩散版本的预训练文件v2-1_512-ema-pruned.ckpt下载并保存到args.pretrained_model_name_or_path="./stable-diffusion/stable-diffusion-2-1-base"路径下。但是在执行下面代码时,总是遇到OSError问题:

python 复制代码
pipeline = DiffusionPipeline.from_pretrained(
            args.pretrained_model_name_or_path,
            torch_dtype=torch_dtype,
            safety_checker=None,
            revision=args.revision,
            )

因此,将args.pretrained_model_name_or_path路径改成了"stabilityai/stable-diffusion-2-1-base"

不足:这种方式需要开放梯子,不然会出现ConnectTimeout

方案二:能否将需要的文件下载到本地,方便执行

1. 下载目标:

代码需要从 Hugging Face 下载 stabilityai/stable-diffusion-2-1-base 模型文件。

相关推荐
OpalStag5810 小时前
MusePublic Art Studio多场景:短视频封面/直播背景/AR滤镜素材生成
stable diffusion·短视频制作·内容创作·ai图像生成
日光明媚11 小时前
torch.compile 与 Triton 的加速本质:从原理到实际效果
人工智能·python·计算机视觉·stable diffusion·aigc
小贺儿开发2 天前
Unity3D 本地 Stable Diffusion 文生图效果演示
人工智能·unity·stable diffusion·文生图·ai绘画·本地化
宁南山9 天前
比迪丽LoRA模型AE做片段视频素材生成:动态背景与元素创作
stable diffusion·ai绘画·lora模型·视频素材生成
日光明媚12 天前
DMD 一步扩散核心原理:从符号定义到梯度推导
人工智能·机器学习·计算机视觉·ai作画·stable diffusion·aigc
baidu_huihui13 天前
ComfyUI 是一款专为 Stable Diffusion 打造的开源可视化操作工具
stable diffusion
李昊哲小课15 天前
Stable Diffusion 保姆级教程|HF 国内镜像配置 + 一键安装 + 本地模型加载 + 完整代码
stable diffusion·hugging face
yumgpkpm15 天前
Qwen3.6正式开源,华为昇腾910B实现高效适配
华为·ai作画·stable diffusion·开源·ai写作·llama·gpu算力
数智工坊16 天前
深度拆解AnomalyAny:异常检测新工作,利用Stable Diffusion生成真实多样异常样本!
人工智能·pytorch·python·stable diffusion
语戚20 天前
深度解析:Stable Diffusion 底层原理 + U-Net Denoise 去噪机制全拆解
人工智能·ai·stable diffusion·aigc·模型