Refactor get_pipe function to handle different request types and pipeline configurations

src/tasks/images/sd.py

@@ -111,14 +111,22 @@ def get_pipe(request: SDReq | SDImg2ImgReq | SDInpaintReq):
             if request.controlnet_config:
                 pipe_args["controlnet"] = controlnet
 
-            if isinstance(request, SDReq):
-                pipe_args['pipeline'] = AutoPipelineForText2Image.from_pipe(**pipe_args)
-            elif isinstance(request, SDImg2ImgReq):
-                pipe_args['pipeline'] = AutoPipelineForImage2Image.from_pipe(**pipe_args)
-            elif isinstance(request, SDInpaintReq):
-                pipe_args['pipeline'] = AutoPipelineForInpainting.from_pipe(**pipe_args)
+            if not request.photomaker_images:
+                if isinstance(request, SDReq):
+                    pipe_args['pipeline'] = AutoPipelineForText2Image.from_pipe(**pipe_args)
+                elif isinstance(request, SDImg2ImgReq):
+                    pipe_args['pipeline'] = AutoPipelineForImage2Image.from_pipe(**pipe_args)
+                elif isinstance(request, SDInpaintReq):
+                    pipe_args['pipeline'] = AutoPipelineForInpainting.from_pipe(**pipe_args)
+                else:
+                    raise ValueError(f"Unknown request type: {type(request)}")
+            elif isinstance(request, any([PhotoMakerStableDiffusionXLPipeline, PhotoMakerStableDiffusionXLControlNetPipeline])):
+                if request.controlnet_config:
+                    pipe_args['pipeline'] = PhotoMakerStableDiffusionXLControlNetPipeline.from_pipe(**pipe_args)
+                else:
+                    pipe_args['pipeline'] = PhotoMakerStableDiffusionXLPipeline.from_pipe(**pipe_args)
             else:
-                raise ValueError(f"Unknown request type: {type(request)}")
+                raise ValueError(f"Invalid request type: {type(request)}")
         
     return pipe_args