Update inference_manager.py
Browse files- inference_manager.py +2 -2
inference_manager.py
CHANGED
@@ -242,8 +242,8 @@ class InferenceManager:
|
|
242 |
|
243 |
build_time = round(time.time() - start, 2)
|
244 |
print(f"Pipeline built with LoRAs in {build_time}s.")
|
245 |
-
if not
|
246 |
-
|
247 |
# Define samplers
|
248 |
samplers = {
|
249 |
"Euler a": EulerAncestralDiscreteScheduler.from_config(temp_pipeline.scheduler.config),
|
|
|
242 |
|
243 |
build_time = round(time.time() - start, 2)
|
244 |
print(f"Pipeline built with LoRAs in {build_time}s.")
|
245 |
+
if not sampler:
|
246 |
+
sampler = self.cfg.get("sampler", "DPM2 a")
|
247 |
# Define samplers
|
248 |
samplers = {
|
249 |
"Euler a": EulerAncestralDiscreteScheduler.from_config(temp_pipeline.scheduler.config),
|