说明文档

StableDiffusion 2.1 转换为 fp16 ONNX 模型，用于运行 GPU 加速的 NodeJS 代码 https://github.com/dakenf/stable-diffusion-nodejs

import * as tf from \"@tensorflow/tfjs-node\"
import { StableDiffusionPipeline } from 'stable-diffusion-nodejs'

const pipe = await StableDiffusionPipeline.fromPretrained(
  'directml', // can be 'cuda' on linux or 'cpu' on mac os
  'aislamov/stable-diffusion-2-1-onnx', // relative path or huggingface repo with onnx model
)

const image = await pipe.run(\"A photo of a cat\", undefined, 1, 9, 30)
const png = await tf.node.encodePng(image[0])
fs.writeFileSync(\"output.png\", png);

aislamov/stable-diffusion-2-1-base-onnx

作者 aislamov

↓ 0 ♥ 5

创建时间: 2023-05-01 00:48:35+00:00

更新时间: 2023-11-07 09:45:48+00:00

在 Hugging Face 上查看

文件 (14)

.DS_Store

.gitattributes

README.md

model_index.json

scheduler/scheduler_config.json

text_encoder/model.onnx ONNX

tokenizer/merges.txt

tokenizer/special_tokens_map.json

tokenizer/tokenizer_config.json

tokenizer/vocab.json

unet/model.onnx ONNX

vae_decoder/model.onnx ONNX

vae_decoder/model.onnx_data

vae_encoder/model.onnx ONNX