Spaces:
Running on Zero
Running on Zero
alx lord commited on
Update core/pipelines/sd_image_pipeline.py
Browse files
core/pipelines/sd_image_pipeline.py
CHANGED
|
@@ -270,7 +270,7 @@ class SdImagePipeline(BasePipeline):
|
|
| 270 |
filename = f"gen_{random.randint(1000000, 9999999)}.png"
|
| 271 |
filepath = os.path.join(out_dir, filename)
|
| 272 |
img.save(filepath, "PNG", pnginfo=metadata)
|
| 273 |
-
|
| 274 |
from utils.cloud_storage import queue_image
|
| 275 |
queue_image(model_type, model_display_name, filepath, params_string)
|
| 276 |
except Exception as _e:
|
|
|
|
| 270 |
filename = f"gen_{random.randint(1000000, 9999999)}.png"
|
| 271 |
filepath = os.path.join(out_dir, filename)
|
| 272 |
img.save(filepath, "PNG", pnginfo=metadata)
|
| 273 |
+
try:
|
| 274 |
from utils.cloud_storage import queue_image
|
| 275 |
queue_image(model_type, model_display_name, filepath, params_string)
|
| 276 |
except Exception as _e:
|