Update core/pipelines/sd_image_pipeline.py

core/pipelines/sd_image_pipeline.py

@@ -270,6 +270,11 @@ class SdImagePipeline(BasePipeline):
                 filename = f"gen_{random.randint(1000000, 9999999)}.png"
                 filepath = os.path.join(out_dir, filename)
                 img.save(filepath, "PNG", pnginfo=metadata)
+                                try:
+                    from utils.cloud_storage import queue_image
+                    queue_image(model_type, model_display_name, filepath, params_string)
+                except Exception as _e:
+                    print(f"Cloud storage skipped: {_e}")
                 final_results.append(filepath)
             
             results = final_results