ssube's picture
add onnx conversion of stabilityai/stable-diffusion-x4-upscaler@19b610c
cc4d9dc verified
|
raw
history blame
819 Bytes
metadata
license: openrail++
tags:
  - stable-diffusion
  - text-to-image
inference: false

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