diffusers-workflow/workflows/templates/upscale-diffusion.json at master · dkackman/diffusers-workflow · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
{
  "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"
      }
    }
  ]
}