安装AI图片生成器(可选)


注意:生成器是可选安装的

生成器要求您的计算机至少具有Nvidia RTX3050以上性能的显卡,8GB或更多的显存显卡。如果您没有满足要求的显卡,请不要安装生成器,使用API即可。

注意:请确保在安装前正确安装了nvidia显卡驱动程序

1. 安装 Python 3.11

  • 对于Windows系统,请在windows应用商店搜索python3.11安装。

  • 对于Linux系统,请自行搜索各系统安装方法,或者参考官方网站。 https://www.python.org/downloads/

2. 下载生成器的安装包

https://github.com/tonera/fluxsd/releases 下载最新的生成器安装包:generator.zip并解压缩。

Linux or Mac
wget https://github.com/tonera/fluxsd/releases/download/{version}/generator.zip
unzip generator.ip
cd generator
Windows

https://github.com/tonera/fluxsd/releases 下载generator.zip并解压缩该文件。然后,在安装包文件夹下右键打开PowerShell的终端窗口。

cd generator

3. 激活环境并安装生成器

Linux系统(pip或者pip3 )

cd fluxsd/generator
python -m venv .venv
. .venv/bin/activate
pip install -r requirement.txt -i https://mirrors.aliyun.com/pypi/simple

如果是windows系统,则执行以下命令 (如果pip可执行文件不存在,请尝试使用pip3

cd fluxsd/generator
python -m venv .venv
.venv/Scripts/activate
pip install -r requirement.txt -i https://mirrors.aliyun.com/pypi/simple

你有可能会遇到无法加载 .ps1 文件的错误提示,这是因为Windows的安全策略阻止了脚本运行,你需要以管理员身份执行

Set-ExecutionPolicy Unrestricted

来解除此限制。

注意:因为其中一个软件包的版本过低,所以请务必修改生成器目录下

.venv\Lib\site-packages\basicsr\data\degradations.py

将第8行

from torchvision.transforms.functional_tensor import rgb_to_grayscale

改为:

from torchvision.transforms.functional import rgb_to_grayscale

4. 安装Torch

注意: 只支持Nvidia系列显卡

打开 https://pytorch.org/ 并按照自己的实际情况选择相应的配置

例如我的显卡CUDA驱动是12.4,则执行以下命令安装

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

如何查看自己的CUDA版本?

nvidia-smi

5. 配置

进入Fluxsd系统,打开配置页,复制配置 -> 图片模型 -> 本地 -> 图片生成器配置文件,将文本框的内容复制到生成器config目录下mk_config.ini文件中

6. 启动生成器

python Service_atz2.py base

如果有错误提示,请按提示操作。

请注意,每次启动生成器时,都需要先激活python环境再执行此脚本。

cd generator
. .venv/bin/activate
python Service_atz2.py base

Windows需要把

. .venv/bin/activate

换为

.venv/Scripts/activate

7. 下载第一个图片模型

模型下载地址: https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_base_1.0.safetensors

将下载下来的文件:sd_xl_base_1.0.safetensors 保存在生成器根目录的models目录下。

现在,打开http://{你的IP地址}:8000你可以开始利用AI生成器在本地生成图片了。

开始使用

FAQ

  1. Windows下执行生成器脚本遇到错误:

    INFO OSError: [WinError 126] The specified module could not be found...“...\lib\site-packages\torch\lib\fbgemm.dll"

    https://www.dllme.com/dll/files/libomp140_x86_64/00637fe34a6043031c9ae4c6cf0a891d/download 下载 libomp140.x86_64.dll 文件保存到/windows/system32目录下。

  2. 错误:“numpy.core._exceptions.MemoryError: Unable to allocate 1.04 MiB for an array with shape (370, 370) and data type float64” 内存不足,请增加虚拟内存空间大小。