在 Mac M1 的 GPU 上运行Stable-Diffusion

Stable Diffusion 是开源的,所以任何人都可以运行和修改它。这就是其在开源之后引发了大量创作热潮的原因

您可以在 Replicate 云中运行 Stable Diffusion,但也可以在本地运行它。除了生成预测之外,您还可以对其进行破解、修改和构建新事物。让它在 M1 Mac 的 GPU 上运行有点繁琐,所以我们创建了本指南来向您展示如何做到这一点。

这一切归功于为GitHub 上的Stable-Diffusion做出贡献的每个人,并在这个 GitHub Issue中解决了所有问题。我们只是他们伟大工作的使者。

我们在之前的工作之上做了一件事:使用 pip 而不是 Conda 来安装依赖项。因为它更容易设置并且不需要编译任何东西。

先决条件

  • 带有 M1 或 M2 芯片的 Mac。
  • 16GB RAM 或更多。8GB 的​​ RAM 可以工作,但速度极慢。
  • macOS 12.3 或更高版本。

设置 Python

您需要 Python 3.10 才能运行稳定扩散。运行python -V以查看您安装的 Python 版本:

$ python3 -V                                                                                       !11338
Python 3.10.6

如果它是 3.10 或更高版本,就像这里一样,你很高兴!跳到下一步。

否则,您需要安装 Python 3.10。最简单的方法是使用 Homebrew。首先,如果您还没有安装 Homebrew 。

然后,安装最新版本的 Python:

brew update
brew install python

现在如果你运行python3 -V你应该有 3.10 或更高版本。您可能需要重新打开控制台才能使其正常工作。

克隆存储库并安装依赖项

运行这个来克隆 Stable Diffusion 的分支:

git clone -b apple-silicon-mps-support https://github.com/bfirsh/stable-diffusion.git
cd stable-diffusion
mkdir -p models/ldm/stable-diffusion-v1/

然后,设置一个 virtualenv 来安装依赖项:

python3 -m pip install virtualenv
python3 -m virtualenv venv

激活虚拟环境:

source venv/bin/activate

(您需要在想要运行稳定扩散的任何时候再次运行此命令。)

然后,安装依赖项:

pip install -r requirements.txt

如果您看到Failed building wheel for onnx可能需要安装这些软件包的错误:

brew install Cmake protobuf rust

下载权重

转到Hugging Face 存储库,阅读并理解许可证,然后单击“访问存储库”。

在该页面上下载sd-v1-4.ckpt(~4 GB)并将其保存models/ldm/stable-diffusion-v1/model.ckpt在您在上面创建的目录中。

运行!

现在,您可以运行稳定扩散:

python scripts/txt2img.py \
  --prompt "a red juicy apple floating in outer space, like a planet" \
  --n_samples 1 --n_iter 1 --plms

你的输出在outputs/txt2img-samples/. 而已。

检查

在scripts/txt2img.py中,代码通过以下方式检查设备情况:

def get_device():
    if(torch.cuda.is_available()):
        return 'cuda'
    elif(torch.backends.mps.is_available()):
        return 'mps'
    else:
        return 'cpu'

运行任务后,你可以通过本地的活动监视器查看资源的使用情况:

使用了Mac M1X的GPU来运行任务

下一步

快乐黑客!

原文地址:https://cloud.tencent.com/developer/article/2137045

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐


学习编程是顺着互联网的发展潮流,是一件好事。新手如何学习编程?其实不难,不过在学习编程之前你得先了解你的目的是什么?这个很重要,因为目的决定你的发展方向、决定你的发展速度。
IT行业是什么工作做什么?IT行业的工作有:产品策划类、页面设计类、前端与移动、开发与测试、营销推广类、数据运营类、运营维护类、游戏相关类等,根据不同的分类下面有细分了不同的岗位。
女生学Java好就业吗?女生适合学Java编程吗?目前有不少女生学习Java开发,但要结合自身的情况,先了解自己适不适合去学习Java,不要盲目的选择不适合自己的Java培训班进行学习。只要肯下功夫钻研,多看、多想、多练
Can’t connect to local MySQL server through socket \'/var/lib/mysql/mysql.sock问题 1.进入mysql路径
oracle基本命令 一、登录操作 1.管理员登录 # 管理员登录 sqlplus / as sysdba 2.普通用户登录
一、背景 因为项目中需要通北京网络,所以需要连vpn,但是服务器有时候会断掉,所以写个shell脚本每五分钟去判断是否连接,于是就有下面的shell脚本。
BETWEEN 操作符选取介于两个值之间的数据范围内的值。这些值可以是数值、文本或者日期。
假如你已经使用过苹果开发者中心上架app,你肯定知道在苹果开发者中心的web界面,无法直接提交ipa文件,而是需要使用第三方工具,将ipa文件上传到构建版本,开...
下面的 SQL 语句指定了两个别名,一个是 name 列的别名,一个是 country 列的别名。**提示:**如果列名称包含空格,要求使用双引号或方括号:
在使用H5混合开发的app打包后,需要将ipa文件上传到appstore进行发布,就需要去苹果开发者中心进行发布。​
+----+--------------+---------------------------+-------+---------+
数组的声明并不是声明一个个单独的变量,比如 number0、number1、...、number99,而是声明一个数组变量,比如 numbers,然后使用 nu...
第一步:到appuploader官网下载辅助工具和iCloud驱动,使用前面创建的AppID登录。
如需删除表中的列,请使用下面的语法(请注意,某些数据库系统不允许这种在数据库表中删除列的方式):
前不久在制作win11pe,制作了一版,1.26GB,太大了,不满意,想再裁剪下,发现这次dism mount正常,commit或discard巨慢,以前都很快...
赛门铁克各个版本概览:https://knowledge.broadcom.com/external/article?legacyId=tech163829
实测Python 3.6.6用pip 21.3.1,再高就报错了,Python 3.10.7用pip 22.3.1是可以的
Broadcom Corporation (博通公司,股票代号AVGO)是全球领先的有线和无线通信半导体公司。其产品实现向家庭、 办公室和移动环境以及在这些环境...
发现个问题,server2016上安装了c4d这些版本,低版本的正常显示窗格,但红色圈出的高版本c4d打开后不显示窗格,
TAT:https://cloud.tencent.com/document/product/1340