Save last screenshot

darthdeus · Sep 19, 2023 · aa4dfad · aa4dfad
1 parent f574e75
commit aa4dfad
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/comfy-wgpu/src/renderer.rs b/comfy-wgpu/src/renderer.rs
@@ -1728,7 +1728,12 @@ impl WgpuRenderer {
                     resized.save(name).unwrap();
 
                     if get_frame() > 60 {
-                        resized.save(format!("{}/{}.png", screenshots_dir, example_name)).unwrap();
+                        resized
+                            .save(format!(
+                                "{}/{}.png",
+                                screenshots_dir, example_name
+                            ))
+                            .unwrap();
 
                         let ffmpeg_command = "ffmpeg";
                         let framerate = "30";