hw6

methods.py

@@ -146,8 +146,8 @@ def infer(
     print(use_advanced_controlnet, use_advanced_ip) 
 
 
-    #1. SD 1.5 + Lora
     if use_advanced_controlnet == None and use_advanced_ip == None:
+        print("1. SD 1.5 + Lora")
         pipe = get_lora_sd_pipeline(base_model_name_or_path=model_repo_id, 
                                     dtype=dtype).to(device)                   
         
@@ -160,7 +160,7 @@ def infer(
                 generator=generator).images[0]
 
     elif use_advanced_controlnet != None and use_advanced_ip == None:
-        #2. SD 1.5 + Lora + Controlnet
+        print("SD 1.5 + Lora + Controlnet")
         edges = cv.Canny(np.array(image_upload_cn))
         edges = np.repeat(edges[:, :, None], 3, axis=2)
         edges = Image.fromarray(edges)
@@ -178,7 +178,7 @@ def infer(
  
 
     elif use_advanced_ip != None and use_advanced_controlnet == None:
-        #3. SD 1.5 + Lora + IpAdapter
+        print("SD 1.5 + Lora + IpAdapter")
         pipe = get_lora_sd_pipeline(base_model_name_or_path=model_repo_id,
                                     ip_adapter=True,
                                     dtype=dtype).to(device)
@@ -193,7 +193,7 @@ def infer(
             generator=generator).images[0]
 
     elif use_advanced_ip != None and use_advanced_controlnet != None:
-        #4. SD 1.5 + Lora + IpAdapter + ControlNet
+        print("SD 1.5 + Lora + IpAdapter + ControlNet")
         pipe = get_lora_sd_pipeline(base_model_name_or_path=model_repo_id,
                                 ip_adapter=True,
                                 controlnet=True,