Update inference_manager.py
Browse files- inference_manager.py +2 -0
inference_manager.py
CHANGED
@@ -437,6 +437,7 @@ class ModelManager:
|
|
437 |
face_strength = inference_params.get("face_strength", 0.1)
|
438 |
sampler = inference_params.get("sampler", cfg.get("sampler", ""))
|
439 |
lora_list = inference_params.get("loras", [])
|
|
|
440 |
|
441 |
if not images:
|
442 |
raise Exception(f"face images not provided")
|
@@ -507,6 +508,7 @@ class ModelManager:
|
|
507 |
height = inference_params.get("height", cfg.get("height", 512))
|
508 |
sampler = inference_params.get("sampler", cfg.get("sampler", ""))
|
509 |
lora_list = inference_params.get("loras", [])
|
|
|
510 |
|
511 |
pipe = model.build_pipeline_with_lora(lora_list, sampler, lora_list)
|
512 |
|
|
|
437 |
face_strength = inference_params.get("face_strength", 0.1)
|
438 |
sampler = inference_params.get("sampler", cfg.get("sampler", ""))
|
439 |
lora_list = inference_params.get("loras", [])
|
440 |
+
seed = inference_params.get("seed", 0)
|
441 |
|
442 |
if not images:
|
443 |
raise Exception(f"face images not provided")
|
|
|
508 |
height = inference_params.get("height", cfg.get("height", 512))
|
509 |
sampler = inference_params.get("sampler", cfg.get("sampler", ""))
|
510 |
lora_list = inference_params.get("loras", [])
|
511 |
+
seed = inference_params.get("seed", 0)
|
512 |
|
513 |
pipe = model.build_pipeline_with_lora(lora_list, sampler, lora_list)
|
514 |
|