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
|
@@ -271,7 +271,7 @@ class SdImagePipeline(BasePipeline):
|
|
| 271 |
filepath = os.path.join(out_dir, filename)
|
| 272 |
img.save(filepath, "PNG", pnginfo=metadata)
|
| 273 |
try:
|
| 274 |
-
from
|
| 275 |
queue_image(model_type, model_display_name, filepath, params_string)
|
| 276 |
except Exception as _e:
|
| 277 |
print(f"Cloud storage skipped: {_e}")
|
|
|
|
| 271 |
filepath = os.path.join(out_dir, filename)
|
| 272 |
img.save(filepath, "PNG", pnginfo=metadata)
|
| 273 |
try:
|
| 274 |
+
from core.cloud_storage import queue_image
|
| 275 |
queue_image(model_type, model_display_name, filepath, params_string)
|
| 276 |
except Exception as _e:
|
| 277 |
print(f"Cloud storage skipped: {_e}")
|