AI作画飞入寻常百姓家——stable diffusion初体验

人工智能52

1. 前言

stable_diffusion来了,这个号称是最强的文本生成图片的模型它来了,相比较DAEE等大模型,它能够让我们消费级的显卡也能够实现文本到图像的生成。下面,我们也来试一下。

2. 准备过程

该服务器上必须要有的基础工具anaconda,wget,git和yum。令我吃惊的是,当我看到wget和yum互为前提的安装时,我头都大了。后来使用curl代替wget解决了yum的安装问题。

3. 执行过程

由于有官方详尽的介绍,因此很容易完成项目的搭建。

  1. 按照官方的地址,我们下载完源码。它是基于latent-diffusion而做的。
  2. 创建环境
    conda env create -f environment.yaml
    conda activate ldm
    需要注意的是,里面有两个github的包,可以从github上下载后,放到src下,使用python进行安装,具体可以参考知乎
  3. 额外注意的有以下3个部分:
    (1)该项目至少要在torch>=1.9.0以上的环境中运行,否则会报错 AttributeError: module 'torch' has no attribute 'inference_mode'
    (2)当遇到 ImportError: cannot import name 'autocast' from 'torch' (/root/anaconda3/envs/ldm/lib/python3.8/site-packages/torch/__init__.py)
    问题时,需要注释掉txt2img.py里的import autocast(第14行)以及使用autocast的那一行(第280行),
    (3)还有需要注意的是,如果遇到的问题是 AttributeError:module ‘distutils’ has no attribute 'version,那么需要安装低版本:
    pip uninstall setuptools
    pip install setuptools==59.5.0 //需要比你之前的低
输入验证码查看隐藏内容

扫描二维码关注本站微信公众号 Johngo学长
或者在微信里搜索 Johngo学长
回复 svip 获取验证码
wechat Johngo学长

相关文章
人工智能

浅谈深度学习的落地问题

前言 深度学习不不仅仅是理论创新,更重要的是应用于工程实际。 关于深度学习人工智能落地,已经有有很多的解决方案,不论是电脑端、手机端还是嵌入式端,将已经训练好的神经网络权重在各个平台跑起来,应用起来才...
人工智能

TensorFlow推荐系统(一)

1 前言 我们浏览在各个平台时会发现"为你推荐"功能。比如YouTube推荐爱看的视频,音乐软件为你提供你可能喜欢的音乐等。其实这一功能的背后涉及的原理就是人工智能的推荐系统。今天我们将介绍Tenso...
人工智能

tensorflow与keras学习

1、tensorflow学习(一定要注意版本对应问题) (1)tensorflow、keras和python对应版本安装 ; (2)tensorflowGPU与cuda、cudnn对应版本安装 官网 ...