zhengchong commited on
Commit
f86262e
·
1 Parent(s): d63caf4

feat: Add return_dict parameter to submit_function_flux for enhanced output control

Browse files

- Introduced the `return_dict` parameter in the `submit_function_flux` function, allowing users to specify the output format of the function's results. This change improves flexibility in handling the function's return values.

Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -333,7 +333,8 @@ def submit_function_flux(
333
  mask_image=mask,
334
  num_inference_steps=num_inference_steps,
335
  guidance_scale=guidance_scale,
336
- generator=generator
 
337
  )[0]
338
 
339
  # Post-processing
 
333
  mask_image=mask,
334
  num_inference_steps=num_inference_steps,
335
  guidance_scale=guidance_scale,
336
+ generator=generator,
337
+ return_dict=False,
338
  )[0]
339
 
340
  # Post-processing