Commit
·
ab735b5
1
Parent(s):
80bab12
Refactor flux_helpers.py to add cleanup in finally block
Browse files
modules/helpers/flux_helpers.py
CHANGED
@@ -119,9 +119,9 @@ def gen_img(request: BaseReq | BaseImg2ImgReq | BaseInpaintReq):
|
|
119 |
if request.refiner:
|
120 |
images = refiner(image=images, prompt=request.prompt, num_inference_steps=40, denoising_start=0.7).images
|
121 |
|
122 |
-
cleanup(pipeline, request.loras)
|
123 |
-
|
124 |
return images
|
125 |
except Exception as e:
|
126 |
cleanup(pipeline, request.loras)
|
127 |
raise gr.Error(f"Error: {e}")
|
|
|
|
|
|
119 |
if request.refiner:
|
120 |
images = refiner(image=images, prompt=request.prompt, num_inference_steps=40, denoising_start=0.7).images
|
121 |
|
|
|
|
|
122 |
return images
|
123 |
except Exception as e:
|
124 |
cleanup(pipeline, request.loras)
|
125 |
raise gr.Error(f"Error: {e}")
|
126 |
+
finally:
|
127 |
+
cleanup(pipeline, request.loras)
|