ONNX Model Zoo

ssube/stable-diffusion-x4-upscaler-onnx

By ssube · ↓ 0 · ♥ 11 · Updated 2023-01-31

What's in this repo

This page lists the files in ssube/stable-diffusion-x4-upscaler-onnx, including model.onnx and more.

  • 3 ONNX models (.onnx files)
  • 13 files, including 6 .json, 3 .onnx, 1 .md, 1 .txt, 1 .pb
  • Last updated 2023-01-31

ONNX files

File Size Download
text_encoder/model.onnx — Download
unet/model.onnx — Download
vae/model.onnx — Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

Stable Diffusion x4 ONNX

This model card focuses on the model associated with the Stable Diffusion Upscaler, available here. This model is trained for 1.25M steps on a 10M subset of LAION containing images >2048x2048. The model was trained on crops of size 512x512 and is a text-guided latent upscaling diffusion model. In addition to the textual input, it receives a noise_level as an input parameter, which can be used to add noise to the low-resolution input according to a predefined diffusion schedule.

Converted to ONNX format for https://github.com/huggingface/diffusers/pull/2158

Model info

onnx stable-diffusion text-to-image arxiv:2112.10752 license:openrail++ region:us
↓ 0 ♥ 11

Created: 2023-01-31

Updated: 2023-01-31

View on Hugging Face

Files (13)

.gitattributes
README.md
low_res_scheduler/scheduler_config.json
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
unet/weights.pb
vae/model.onnx ONNX

Related models