an update to the original AIO version of this workflow.

only works for all-in-one flux safetensor checkpoints