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 模型文件。

相关推荐
学易10 天前
第二十节.探索新技术:如何自学SD3模型(上)(找官方资料/精读/下载/安装/3款工作流/效果测试)
人工智能·ai作画·stable diffusion·comfyui·工作流·sd3
Bachnroth11 天前
Stable Diffusion 3.5 FP8镜像负载均衡配置建议
stable diffusion· fp8· 负载均衡
新缸中之脑11 天前
Stable Diffusion的3个替代方案
人工智能·stable diffusion
2401_8288906414 天前
实现扩散模型 Stable Diffusion - MNIST 数据集
人工智能·python·深度学习·stable diffusion
凯子坚持 c16 天前
在 openJiuwen 里把在线小工具搬回本地
人工智能·windows·stable diffusion·openteledb·openclaw
空白诗24 天前
CANN ops-nn 算子解读:Stable Diffusion 图像生成中的 Conv2D 卷积实现
深度学习·计算机视觉·stable diffusion
学易24 天前
第十五节.别人的工作流,如何使用和调试(上)?(2类必现报错/缺失节点/缺失模型/思路/实操/通用调试步骤)
人工智能·ai作画·stable diffusion·报错·comfyui·缺失节点
心疼你的一切24 天前
基于CANN仓库算力手把手实现Stable Diffusion图像生成(附完整代码+流程图)
数据仓库·深度学习·stable diffusion·aigc·流程图·cann
Niuguangshuo25 天前
DALL-E 3:如何通过重构“文本描述“革新图像生成
人工智能·深度学习·计算机视觉·stable diffusion·重构·transformer
Niuguangshuo1 个月前
深入解析 Stable Diffusion XL(SDXL):改进潜在扩散模型,高分辨率合成突破
stable diffusion