WasmEdge Stable Diffusion

当Wasi在WasmEdge上执行时，用于稳定扩散函数的Rust库。

设置WasmEdge

git clone https://github.com/WasmEdge/WasmEdge.git
cd WasmEdge
cmake -GNinja -Bbuild -DCMAKE_BUILD_TYPE=Release -DWASMEDGE_BUILD_TESTS=OFF -DWASMEDGE_PLUGIN_STABLEDIFFUSION=On
cmake --build build
cmake --install build

下载模型

从以下命令下载权重或量化模型。
您还可以使用我们的示例自行量化权重。

stable-diffusion v1.4: second-state/stable-diffusion-v-1-4-GGUF
stable-diffusion v1.5: second-state/stable-diffusion-v1-5-GGUF
stable-diffusion v2.1: second-state/stable-diffusion-2-1-GGUF

curl -L -O https://hugging-face.cn/second-state/stable-diffusion-v-1-4-GGUF/resolve/main/sd-v1-4.ckpt

curl -L -O https://hugging-face.cn/second-state/stable-diffusion-v-1-4-GGUF/resolve/main/stable-diffusion-v1-4-Q8_0.gguf

运行示例

cargo build --target wasm32-wasi --release
wasmedge --dir .:. ./target/wasm32-wasi/release/wasmedge_stable_diffusion_example.wasm

然后您将看到三个新文件。

1. sd-v1-4-Q8_0.gguf: 是sd-v1-4的量化
2. output.png: 一张有猫的图片
3. output2.png: 一张有蓝眼睛的猫的图片。