Update utils.py
Browse files
utils.py
CHANGED
|
@@ -179,6 +179,4 @@ def load_pipeline(model_name, device, hf_token=None, vae=None):
|
|
| 179 |
add_watermarker=False
|
| 180 |
)
|
| 181 |
return pipe.to(device)
|
| 182 |
-
|
| 183 |
-
logging.error(f"Failed to load pipeline: {str(e)}", exc_info=True)
|
| 184 |
-
raise
|
|
|
|
| 179 |
add_watermarker=False
|
| 180 |
)
|
| 181 |
return pipe.to(device)
|
| 182 |
+
|
|
|
|
|
|