StableDiffusion本地启动全攻略

作者:Nicky2024.11.20 19:05浏览量:69

简介:本文介绍了在无法连接HuggingFace时,如何手动下载StableDiffusion的模型和参数文件,并修改项目代码以适应本地文件,成功启动StableDiffusion项目的详细步骤。

在AI绘图领域,StableDiffusion无疑是一个炙手可热的话题。然而,对于许多初学者来说,由于网络问题无法连接到HuggingFace,导致项目启动失败,成为了一个不小的困扰。本文将详细介绍在无法连接HuggingFace的情况下,如何手动下载StableDiffusion的模型和参数文件,并成功启动项目。

一、下载安装git与StableDiffusion项目

首先,你需要确保电脑上已经安装了git。git是一款开源的分布式版本控制系统,用于高效地处理任何大小的项目。你可以通过搜索引擎找到git的官方网站,并按照提示进行下载安装。

接下来,访问StableDiffusion的GitHub页面:https://github.com/AUTOMATIC1111/stable-diffusion-webui。使用git将项目clone到本地,或者直接下载压缩包解压。下载完成后,你会得到一个包含StableDiffusion项目文件的文件夹。

二、手动下载模型与参数文件

由于无法连接到HuggingFace,我们需要手动下载所需的模型和参数文件。模型文件通常包括.ckpt和.safetensors两种格式,而参数文件则包括vocab.json、tokenizer_config.json、special_tokens_map.json、merges.txt和config.json等。

你可以从一些可靠的第三方网站(如互链高科aliendao.cn)下载这些文件。下载完成后,将模型文件放入项目的\models\Stable-diffusion目录中,将参数文件放入项目目录下的.cache\huggingface\openai\clip-vit-large-patch14文件夹中。

三、修改项目代码

接下来,我们需要修改项目代码以适应本地文件。打开repositories/stable-diffusion-stability-ai/ldm/modules/encoders/modules.py文件,找到第100行的init方法。将version参数的值从’openai/clip-vit-large-patch14’改为你的参数文件保存目录。

四、启动项目

双击webui.bat或webui-user.bat文件尝试启动项目。由于我们的模型和参数文件是从本地加载的,可能会遇到检查不通过的情况。此时,你需要在命令行中添加—disable-safe-unpickle参数以绕过检查。

打开cmd命令行工具,切换到项目目录,执行以下命令:

  1. webui.bat --disable-safe-unpickle

注意,如果你使用webui-user.bat启动项目,需要确保该文件中也包含了—disable-safe-unpickle参数。否则,你可以直接修改webui-user.bat文件,在适当的位置添加该参数。

五、问题解决与替代方案

如果在启动过程中遇到版本兼容性问题(如ImportError: cannot import name ‘_compare_version’ from ‘torchmetrics.utilities.imports’),你可以尝试降低torchmetrics等库的版本。例如,使用pip安装指定版本的torchmetrics、torch等库:

  1. !pip install torchmetrics==0.11.4 torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchtext==0.14.1 torchaudio==0.13.1 torchdata==0.5.1 --extra-index-url https://download.pytorch.org/whl/cu117

此外,你还可以尝试使用HuggingFace的镜像网站(如hf-mirror.com)来访问HuggingFace的资源和功能。这些镜像网站通常提供一个备用的域名来绕过访问限制。

六、产品关联:千帆大模型开发与服务平台

在StableDiffusion项目的实际应用中,你可能会遇到模型优化、部署和集成等挑战。此时,千帆大模型开发与服务平台可以为你提供全方位的支持。该平台提供了丰富的模型库和工具链,可以帮助你轻松实现模型的训练、优化和部署。同时,千帆大模型开发与服务平台还支持与StableDiffusion等主流AI绘图工具的集成,让你的项目更加高效和便捷。

总之,通过手动下载模型和参数文件并修改项目代码,我们可以成功启动StableDiffusion项目。同时,借助千帆大模型开发与服务平台等工具的支持,我们可以更好地应用StableDiffusion进行AI绘图创作。