test p2p

app_haircolor_img2img.py

@@ -44,12 +44,10 @@ pidiNet_detector = pidiNet_detector.to(DEVICE)
 hed_detector = HEDdetector.from_pretrained('lllyasviel/Annotators')
 hed_detector = hed_detector.to(DEVICE)
 
-controlnet = [
-    ControlNetModel.from_pretrained(
-        "lllyasviel/control_v11e_sd15_ip2p", 
-        torch_dtype=torch.float16,
-    ),
-]
+controlnet = ControlNetModel.from_pretrained(
+    "lllyasviel/control_v11e_sd15_ip2p", 
+    torch_dtype=torch.float16,
+)
 
 basepipeline = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
     BASE_MODEL,
@@ -77,7 +75,7 @@ def image_to_image(
     time_cost_str = ''
     run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
 
-    cond_image = [input_image]
+    cond_image = input_image
 
     # generator = torch.Generator(device=DEVICE).manual_seed(seed)
     generated_image = basepipeline(