barreloflube commited on
Commit
a7c9e18
·
1 Parent(s): 5dc606d

Refactor scheduler loading in get_pipe function

Browse files
Files changed (1) hide show
  1. 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 = get_scheduler(pipe_args["pipeline"], request.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: