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
Created: 2023-01-31
Updated: 2023-01-31
View on Hugging FaceFiles (13)
Related models
More from ssube
No other models by this author.