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

Refactor flux_helpers.py to update pipeline mode based on request type

Browse files
Files changed (1) hide show
  1. modules/helpers/flux_helpers.py +4 -4
modules/helpers/flux_helpers.py CHANGED
@@ -41,12 +41,12 @@ def get_pipe(request: BaseReq | BaseImg2ImgReq | BaseInpaintReq):
41
  pipe_args["controlnet"] = [controlnet]
42
 
43
  # Choose Pipeline Mode
44
- if isinstance(request, BaseReq):
45
- pipe_args['pipeline'] = AutoPipelineForText2Image.from_pipe(**pipe_args)
46
  elif isinstance(request, BaseImg2ImgReq):
47
  pipe_args['pipeline'] = AutoPipelineForImage2Image.from_pipe(**pipe_args)
48
- elif isinstance(request, BaseInpaintReq):
49
- pipe_args['pipeline'] = AutoPipelineForInpainting.from_pipe(**pipe_args)
50
 
51
  # Enable or Disable Refiner
52
  if request.vae:
 
41
  pipe_args["controlnet"] = [controlnet]
42
 
43
  # Choose Pipeline Mode
44
+ if isinstance(request, BaseInpaintReq):
45
+ pipe_args['pipeline'] = AutoPipelineForInpainting.from_pipe(**pipe_args)
46
  elif isinstance(request, BaseImg2ImgReq):
47
  pipe_args['pipeline'] = AutoPipelineForImage2Image.from_pipe(**pipe_args)
48
+ elif isinstance(request, BaseReq):
49
+ pipe_args['pipeline'] = AutoPipelineForText2Image.from_pipe(**pipe_args)
50
 
51
  # Enable or Disable Refiner
52
  if request.vae: