Fooocus：一个简单且功能强大的Stable Diffusion webUI

news2024/10/7 6:50:58

Stable Diffusion是一个强大的图像生成AI模型，但它通常需要大量调整和提示工程。Fooocus的目标是改变这种状况。

Fooocus的创始人Lvmin Zhang（也是 ControlNet论文的作者）将这个项目描述为对“Stable Diffusion”和“ Midjourney”设计的重新设计。Fooocus就像是Midjourney的免费离线版本，但是它使用SDXL模型。或者说它将Stable Diffusion的出图过程做了非常好的优化，没有那么多繁琐的配置。

Fooocus内置和自动化了许多优化和质量改进，将其他页面的手动设置变为了自动配置，这样就和Midjourney一样，这将在每次尝试中获得良好的结果。如果你想做更多，你可以使用Fooocus的Advanced选项卡。比如说设置清晰度过滤器或自定义lora。

在这篇文章中，我们将介绍如何在本地和Colab上使用Fooocus

在Windows上运行

只需要在下载文件，解压缩，然后运行run.bat,就是这么简单

在第一次运行时，它会自动下载模型，如果你已经有这些文件，你可以把它们复制到上面的位置，以加快安装速度。

sd_xl_base_1.0_0.9vae.safetensors
sd_xl_refiner_1.0_0.9vae.safetensors

Fooocus可以在16gb RAM和6GB VRAM的系统上运行，性能非常好，下面图片来自的Github。

最低要求是4GB Nvidia GPU内存(4GB VRAM)和8GB系统内存(8GB RAM)。

在Linux上运行

Linux上会更加简单：

 git clone https://github.com/lllyasviel/Fooocus.git
 cd Fooocus
 conda env create -f environment.yaml
 conda activate fooocus
 pip install -r requirements_versions.txt

与Windows类似下载模型以加快过程，但是他的启动命令变为:

 python launch.py

或者如果想打开远程端口，需要使用listen参数

 python launch.py --listen

在Google Colab上运行

因为需要使用GPU，所以我们这里选择T4 GPU，因为它已经够用了

然后使用下面命令，由于下载和安装，该操作可能需要一些时间才能完成，但是Colab的下载速度很快，我们不需要传输模型了

 %cd /content
 !git clone https://github.com/lllyasviel/Fooocus
 
 !apt -y update -qq
 !wget https://github.com/camenduru/gperftools/releases/download/v1.0/libtcmalloc_minimal.so.4 -O /content/libtcmalloc_minimal.so.4
 %env LD_PRELOAD=/content/libtcmalloc_minimal.so.4
 
 !pip install torchsde==0.2.5 einops==0.4.1 transformers==4.30.2 safetensors==0.3.1 accelerate==0.21.0
 !pip install pytorch_lightning==1.9.4 omegaconf==2.2.3 gradio==3.39.0 xformers==0.0.20 triton==2.0.0 pygit2==1.12.2
 
 !apt -y install -qq aria2
 !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/sd_xl_base_1.0/resolve/main/sd_xl_base_1.0_0.9vae.safetensors -d /content/Fooocus/models/checkpoints -o sd_xl_base_1.0_0.9vae.safetensors
 !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/ckpt/sd_xl_refiner_1.0/resolve/main/sd_xl_refiner_1.0_0.9vae.safetensors -d /content/Fooocus/models/checkpoints -o sd_xl_refiner_1.0_0.9vae.safetensors
 !aria2c --console-log-level=error -c -x 16 -s 16 -k 1M https://huggingface.co/stabilityai/stable-diffusion-xl-base-1.0/resolve/main/sd_xl_offset_example-lora_1.0.safetensors -d /content/Fooocus/models/loras -o sd_xl_offset_example-lora_1.0.safetensors
 
 %cd /content/Fooocus
 !git pull
 !python launch.py --share

当它完成时，会看到一个连接，类似下图