StabilityAI Suite - Creative Upscale Recover File

Creative Upscale runs in two steps: generation, then your machine pulls the finished image back. Step one happens on Stability's servers; step two happens on your PC. If your PC crashes, loses internet, or you close ComfyUI between the two, the second step never completes - and because you already paid for that generation, losing the image is doubly annoying.

This node is the do-over for exactly that situation. Every time Creative Upscale kicks off, the pack appends a line to log.txt in its own folder recording the job's unique ID. Stability keeps the finished image for 24 hours. Paste that ID in here and the node fetches the image back.

How it works

It's a plain GET against stable-image/upscale/creative/result/<id>, polling every 10 seconds until the result is ready, then decoding the returned base64 into a normal IMAGE. Nothing fancy - and that's the point. It's a rescue tool, not a workflow component.

Inputs and output

One input: image_id, the string you copied from log.txt. Open that file, find the ID of the failed run, paste it in. The output is image_out, an IMAGE - hook it to a Save Image node and you're done.

One gotcha specific to this node: it uses the API key from the environment variable or sai_platform_key.txt, and there's no api_key_override field here. If you normally rely on the per-node override, make sure SAI_API_KEY is set before you need this node, or the rescue fails at the door.

Also be aware the clock starts the moment the original generation finished. Stability archives images for 24 hours - after that the ID returns nothing and your credits are gone for good. Don't wait.

Install

Same as the rest of the pack - ComfyUI Manager (search "ComfyUI StabilityAI Suite") or:

cd ComfyUI/custom_nodes
git clone https://github.com/florestefano1975/ComfyUI-StabilityAI-Suite

Restart ComfyUI, set your API key, done. No models, no extra dependencies - the whole pack is a thin wrapper over the Stability API.

Honest take: if you never crash and never lose internet, you'll never touch this node. It's insurance, and cheap insurance at that - one field, one run, and it can claw back a paid generation you'd otherwise have to re-run (and re-pay) from scratch.