Update inference_manager.py
Browse files- 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", "
|
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),
|