当前位置: 首页 > article >正文

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

联网下载没有问题:

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问题:

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


http://www.kler.cn/a/371900.html

相关文章:

  • net mvc中使用vue自定义组件遇到的坑
  • 深度了解flink(八) JobManager(2)initializeServices剖析
  • 芯片上音频相关的验证
  • SpringCloud笔记
  • Java应用程序的测试覆盖率之设计与实现(四)-- jacoco-maven-plugin
  • 【Oracle实验】字段为空的,无法通过排除判断
  • 网络爬虫的基本原理是什么?
  • 初始Docker
  • NVR设备ONVIF接入平台EasyCVR视频分析设备平台视频质量诊断技术与能力
  • 深入解析 MySQL 数据库:数据类型
  • Rust精简核心笔记:第二波,语法精髓部分解锁
  • 十六:Python学习笔记-- 爬虫(2)requests 模块详解
  • 装饰器怎样实现
  • LeetCode --- 420周赛
  • 大数据新视界 -- 大数据大厂之大数据重塑影视娱乐产业的未来(4 - 3)
  • linux查看系统负载情况
  • STM32--I2C外设
  • Java AQS Semaphore 源码
  • Jenkins面试整理-什么是 Jenkins?
  • kettle8.3 Oracle连接运行一段时间后:Socket read timed out
  • ClickHouse 3节点集群安装
  • 香橙派Orangepi 5plus 配置Hailo-8/Hailo-8L
  • mariadb数据库中文乱码问题
  • 微服务之链路追踪Sleuth+zipkin
  • Linux 上使用 Docker 下载和运行 Redis
  • 智能家居10G雷达感应开关模块,飞睿智能uA级别低功耗、超高灵敏度,瞬间响应快