Commit
·
a7c9e18
1
Parent(s):
5dc606d
Refactor scheduler loading in get_pipe function
Browse files- tabs/images/handlers.py +1 -1
tabs/images/handlers.py
CHANGED
@@ -71,7 +71,7 @@ def get_pipe(request: BaseReq | BaseImg2ImgReq | BaseInpaintReq):
|
|
71 |
pipe_args["pipeline"].vae = None if model['loader'] == 'sdxl' else flux_vae
|
72 |
|
73 |
# Set Scheduler
|
74 |
-
pipe_args["pipeline"].scheduler =
|
75 |
|
76 |
# Set Loras
|
77 |
if request.loras:
|
|
|
71 |
pipe_args["pipeline"].vae = None if model['loader'] == 'sdxl' else flux_vae
|
72 |
|
73 |
# Set Scheduler
|
74 |
+
pipe_args["pipeline"].scheduler = load_scheduler(pipe_args["pipeline"], request.scheduler)
|
75 |
|
76 |
# Set Loras
|
77 |
if request.loras:
|