summaryrefslogtreecommitdiff
path: root/candle-examples/examples/flux/main.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-examples/examples/flux/main.rs')
-rw-r--r--candle-examples/examples/flux/main.rs6
1 files changed, 5 insertions, 1 deletions
diff --git a/candle-examples/examples/flux/main.rs b/candle-examples/examples/flux/main.rs
index 943db112..12439892 100644
--- a/candle-examples/examples/flux/main.rs
+++ b/candle-examples/examples/flux/main.rs
@@ -250,7 +250,11 @@ fn run(args: Args) -> Result<()> {
};
println!("img\n{img}");
let img = ((img.clamp(-1f32, 1f32)? + 1.0)? * 127.5)?.to_dtype(candle::DType::U8)?;
- candle_examples::save_image(&img.i(0)?, "out.jpg")?;
+ let filename = match args.seed {
+ None => "out.jpg".to_string(),
+ Some(s) => format!("out-{s}.jpg"),
+ };
+ candle_examples::save_image(&img.i(0)?, filename)?;
Ok(())
}