Commit
·
a685f13
1
Parent(s):
ab735b5
Refactor flux_helpers.py to update pipeline mode based on request type
Browse files
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,
|
45 |
-
pipe_args['pipeline'] =
|
46 |
elif isinstance(request, BaseImg2ImgReq):
|
47 |
pipe_args['pipeline'] = AutoPipelineForImage2Image.from_pipe(**pipe_args)
|
48 |
-
elif isinstance(request,
|
49 |
-
pipe_args['pipeline'] =
|
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:
|