barreloflube commited on
Commit
52aff8d
·
1 Parent(s): 0ec1b6c

Refactor flux_helpers.py to include pipeline arguments in log message

Browse files
Files changed (1) hide show
  1. modules/helpers/flux_helpers.py +1 -1
modules/helpers/flux_helpers.py CHANGED
@@ -113,7 +113,7 @@ def gen_img(request: BaseReq | BaseImg2ImgReq | BaseInpaintReq):
113
  args['mask_image'] = resize_images([request.mask_image], request.height, request.width, request.resize_mode)[0]
114
 
115
  # Generate
116
- gr.Info(f"Using {pipeline}", duration=150)
117
  images = pipeline(**args).images
118
 
119
  # Refiner
 
113
  args['mask_image'] = resize_images([request.mask_image], request.height, request.width, request.resize_mode)[0]
114
 
115
  # Generate
116
+ gr.Info(f"Using {pipeline}\nArgs:\n{str(args)}", duration=150)
117
  images = pipeline(**args).images
118
 
119
  # Refiner