nsfwalex commited on
Commit
d193d40
1 Parent(s): 20c459f

Update inference_manager.py

Browse files
Files changed (1) hide show
  1. inference_manager.py +3 -3
inference_manager.py CHANGED
@@ -194,7 +194,7 @@ class InferenceManager:
194
  #variant="fp16",
195
  custom_pipeline = "lpw_stable_diffusion_xl",
196
  )
197
- pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
198
  clip_skip = cfg.get("clip_skip", 1)
199
  # Adjust clip skip for XL (assumed not relevant for SD 1.5)
200
  pipe.text_encoder.config.num_hidden_layers -= (clip_skip - 1)
@@ -202,7 +202,7 @@ class InferenceManager:
202
  load_time = round(time.time() - start, 2)
203
  print(f"Base model loaded in {load_time}s")
204
 
205
- if cfg.get("load_ip_adapter_faceid", False):
206
  if model_version in ("pony", "xl"):
207
  ip_ckpt = self.ext_model_pathes.get("ip-adapter-faceid-sdxl", "")
208
  if ip_ckpt:
@@ -284,7 +284,7 @@ class InferenceManager:
284
  build_time = round(time.time() - start, 2)
285
  print(f"Pipeline built with LoRAs in {build_time}s.")
286
  if not sampler:
287
- sampler = self.cfg.get("sampler", "DPM2 a")
288
  # Define samplers
289
  samplers = {
290
  "Euler a": EulerAncestralDiscreteScheduler.from_config(temp_pipeline.scheduler.config),
 
194
  #variant="fp16",
195
  custom_pipeline = "lpw_stable_diffusion_xl",
196
  )
197
+ #pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
198
  clip_skip = cfg.get("clip_skip", 1)
199
  # Adjust clip skip for XL (assumed not relevant for SD 1.5)
200
  pipe.text_encoder.config.num_hidden_layers -= (clip_skip - 1)
 
202
  load_time = round(time.time() - start, 2)
203
  print(f"Base model loaded in {load_time}s")
204
 
205
+ if cfg.get("load_ip_adapter_faceid", False) and False:
206
  if model_version in ("pony", "xl"):
207
  ip_ckpt = self.ext_model_pathes.get("ip-adapter-faceid-sdxl", "")
208
  if ip_ckpt:
 
284
  build_time = round(time.time() - start, 2)
285
  print(f"Pipeline built with LoRAs in {build_time}s.")
286
  if not sampler:
287
+ sampler = self.cfg.get("sampler", "Euler a")
288
  # Define samplers
289
  samplers = {
290
  "Euler a": EulerAncestralDiscreteScheduler.from_config(temp_pipeline.scheduler.config),