vitoplantamura / OnnxStream

Lightweight inference library for ONNX files, written in C++. It can run Stable Diffusion XL 1.0 on a RPI Zero 2 (or in 298MB of RAM) but also Mistral 7B on desktops and servers. ARM, x86, WASM, RISC-V supported. Accelerated by XNNPACK.
https://yolo.vitoplantamura.com/
Other
1.86k stars 84 forks source link

Can we working on solution to save every steps into separated / same PNG files? #38

Open ThomAce opened 1 year ago

ThomAce commented 1 year ago

To see how the diffusion is progressing, it would be good to see diffused image stages in - for example - every single steps. The user could use paramter to show it or not. When showing like --showsteps without any extra parameter would show diffused image from every single steps. I know it would cause some extra cpu overhead, but would be really entertaining on more powerful systems than RPi.

I already started to thinking about how to achieve this, but other more talented developers might know this already.

Update:

I found the way to achieve it.

kaosbeat commented 2 months ago

care to briefly share how you did this?

ThomAce commented 2 months ago

@kaosbeat Yes of course. But please keep in mind that this was only for experimental purposes! image