nsfwalex commited on
Commit
ecc7e3c
·
verified ·
1 Parent(s): 316f7d7

Update inference_manager.py

Browse files
Files changed (1) hide show
  1. 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