upscale-diffusion.json

{
  "id": "DiffusionUpscaleImage",
  "description": "Upscales an input image with a Stable Diffusion upscale pipeline, guided by a prompt. There is no generation step, so any existing image can be upscaled. First load adds about 2 minutes.",
  "cost": [
    {"device": "cuda", "name": "RTX 3090", "vram_gb": 24, "minutes": 0.2}
  ],
  "cost_drivers": ["num_inference_steps"],
  "variables": {
    "input_image": {
      "location": "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png"
    },
    "prompt": "",
    "negative_prompt": null,
    "mode": "x2",
    "num_inference_steps": 25,
    "guidance_scale": 9.0,
    "noise_level": 20,
    "device": "cuda"
  },
  "steps": [
    {
      "name": "upscale",
      "task": {
        "command": "diffusion_upscale",
        "arguments": {
          "image": "variable:input_image",
          "prompt": "variable:prompt",
          "negative_prompt": "variable:negative_prompt",
          "mode": "variable:mode",
          "num_inference_steps": "variable:num_inference_steps",
          "guidance_scale": "variable:guidance_scale",
          "noise_level": "variable:noise_level",
          "device": "variable:device"
        }
      },
      "result": {
        "content_type": "image/png",
        "subfolder": "final"
      }
    }
  ]
}