zhiweili commited on
Commit
11b5282
·
1 Parent(s): 963fadc

fix cond image

Browse files
app_haircolor_inpaint_adapter_15.py CHANGED
@@ -90,7 +90,7 @@ def image_to_image(
90
  # lineart_image = lineart_detector(input_image, int(generate_size*1), generate_size)
91
  # run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
92
  pidiNet_image = pidiNet_detector(input_image, int(generate_size*1), generate_size)
93
- cond_image = [pidiNet_image, canny_image]
94
  cond_scale = [cond_scale1, cond_scale2]
95
 
96
  generator = torch.Generator(device=DEVICE).manual_seed(seed)
 
90
  # lineart_image = lineart_detector(input_image, int(generate_size*1), generate_size)
91
  # run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
92
  pidiNet_image = pidiNet_detector(input_image, int(generate_size*1), generate_size)
93
+ cond_image = [canny_image, pidiNet_image]
94
  cond_scale = [cond_scale1, cond_scale2]
95
 
96
  generator = torch.Generator(device=DEVICE).manual_seed(seed)