说明文档

使用方法：

from diffusers import StableDiffusionOnnxPipeline, PNDMScheduler
model_path = r'ShadowPower/waifu-diffusion-diffusers-onnx-v1-3'

scheduler = PNDMScheduler(
    beta_start=0.00085,
    beta_end=0.012,
    beta_schedule='scaled_linear',
    skip_prk_steps=True,
    tensor_format='np'
)
pipe = StableDiffusionOnnxPipeline.from_pretrained(
    model_path,
    provider=\"CPUExecutionProvider\",
    scheduler=scheduler
)

if __name__ == '__main__':
    prompt = \"1girl,  hakurei reimu\"
    image = pipe(prompt).images[0]

ShadowPower/waifu-diffusion-diffusers-onnx-v1-3

作者 ShadowPower

↓ 0 ♥ 2

创建时间: 2022-10-11 08:46:05+00:00

更新时间: 2022-10-11 13:22:21+00:00

在 Hugging Face 上查看

文件 (15)

.gitattributes

README.md

feature_extractor/preprocessor_config.json

model_index.json

safety_checker/model.onnx ONNX

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

unet/weights.pb

vae_decoder/model.onnx ONNX

vae_encoder/model.onnx ONNX