Stable-Diffusion-XL-Burn 项目使用指南

Stable-Diffusion-XL-Burn 项目使用指南

stable-diffusion-xl-burn Stable Diffusion XL ported to Rust's burn framework 项目地址: https://gitcode.com/gh_mirrors/st/stable-diffusion-xl-burn

1. 项目目录结构及介绍

stable-diffusion-xl-burn/
├── Cargo.lock
├── Cargo.toml
├── LICENSE
├── README.md
├── src/
│   ├── tokenizer/
│   └── ...
├── python/
│   └── ...
├── SDXL1.0/
│   ├── diffuser.mpk
│   ├── diffuser.cfg
│   ├── embedder.mpk
│   ├── embedder.cfg
│   ├── latent_decoder.mpk
│   ├── latent_decoder.cfg
│   └── refiner.mpk (可选)
│   └── refiner.cfg (可选)
└── ...

目录结构说明

• Cargo.lock: Rust 项目的依赖锁定文件。
• Cargo.toml: Rust 项目的配置文件，定义了项目的依赖和元数据。
• LICENSE: 项目的开源许可证文件，本项目使用 MIT 许可证。
• README.md: 项目的介绍和使用说明文档。
• src/: 包含 Rust 项目的源代码文件。
• python/: 包含 Python 脚本，用于模型文件的转换。
• SDXL1.0/: 包含预转换的 Stable Diffusion XL 1.0 模型文件，包括 diffuser.mpk, diffuser.cfg, embedder.mpk, embedder.cfg, latent_decoder.mpk, latent_decoder.cfg，以及可选的 refiner.mpk 和 refiner.cfg。

2. 项目启动文件介绍

启动文件

• src/main.rs: 这是 Rust 项目的主入口文件，负责启动和运行 Stable Diffusion XL 模型。

启动命令

cargo run --release --bin sample -- --model-dir SDXL1.0 --output-dir /path/to/output --prompt "An elegant bright red crab"

命令说明

• --model-dir: 指定模型文件所在的目录，例如 SDXL1.0。
• --output-dir: 指定输出图像的保存路径。
• --prompt: 指定生成图像的文本提示。

3. 项目的配置文件介绍

配置文件

• Cargo.toml: 定义了 Rust 项目的依赖和元数据。

[package]
name = "stable-diffusion-xl-burn"
version = "0.1.0"
edition = "2021"

[dependencies]
burn = "0.1"
...

配置文件说明

• name: 项目的名称。
• version: 项目的版本号。
• edition: 使用的 Rust 版本。
• dependencies: 项目依赖的库，例如 burn 框架。

通过以上步骤，您可以成功启动并使用 Stable-Diffusion-XL-Burn 项目。

stable-diffusion-xl-burn Stable Diffusion XL ported to Rust's burn framework 项目地址: https://gitcode.com/gh_mirrors/st/stable-diffusion-xl-burn