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

相关推荐
平安的平安1 天前
用 Python 玩转 AI 绘图:Stable Diffusion 本地部署指南
人工智能·python·stable diffusion
语戚1 天前
Stable Diffusion 核心模块深度拆解:CLIP、U-Net 与 VAE 原理全解析
人工智能·ai·stable diffusion·aigc·模型
yumgpkpm3 天前
华为昇腾910B上用Kubernetes(K8s)部署LLM和用Docker部署LLM的区别
docker·chatgpt·容器·stable diffusion·kubernetes·llama·gpu算力
语戚3 天前
Stable Diffusion 入门:架构、空间与生成流程概览
人工智能·ai·stable diffusion·aigc·模型
yumgpkpm3 天前
华为昇腾910B上用Kubernetes(K8s)部署LLM(Qwen3-32B)的详细步骤,保姆级命令及方法、下载链接等
运维·服务器·华为·stable diffusion·aigc·copilot·llama
平安的平安3 天前
Python 实现 AI 图像生成:调用 Stable Diffusion API 完整教程
人工智能·python·stable diffusion
孤岛站岗8 天前
【AI大模型入门】B02:Stable Diffusion——开源绘图,让AI绘画飞入寻常百姓家
人工智能·stable diffusion·开源
智星云算力8 天前
从创作到训练:Stable Diffusion WebUI全链路实战
人工智能·深度学习·stable diffusion·智星云·gpu算力租用
ai_xiaogui9 天前
PanelAI 最新进展:AI算力集群一键部署ComfyUI & Stable Diffusion,多节点Docker管理面板从原型到真实数据全解析
人工智能·docker·stable diffusion·ai算力集群管理·ai私有化部署工具·ai容器管理与算力调度实战·多节点docker服务器面板
scott19851210 天前
扩散模型之(二十)Stable Diffusion的运作原理
深度学习·stable diffusion·扩散模型·生成式