barreloflube commited on
Commit
ab735b5
·
1 Parent(s): 80bab12

Refactor flux_helpers.py to add cleanup in finally block

Browse files
Files changed (1) hide show
  1. modules/helpers/flux_helpers.py +2 -2
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)