Kolors-Controlnet_and_IPA

Running on Zero

lixiang46 commited on Aug 2

Commit

5bc0c16

•

1 Parent(s): d72888b

debug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -65,7 +65,9 @@ pipe_canny = StableDiffusionXLControlNetImg2ImgPipeline(
 for pipe in [pipe_depth]:
     if hasattr(pipe.unet, 'encoder_hid_proj'):
         pipe.unet.text_encoder_hid_proj = pipe.unet.encoder_hid_proj
-    pipe.load_ip_adapter(f'{ckpt_dir_ipa}' , subfolder="", weight_name=["ip_adapter_plus_general.bin"])
 @spaces.GPU
 def process_canny_condition(image, canny_threods=[100,200]):

 for pipe in [pipe_depth]:
     if hasattr(pipe.unet, 'encoder_hid_proj'):
         pipe.unet.text_encoder_hid_proj = pipe.unet.encoder_hid_proj
+pipe_depth.load_ip_adapter(f'{ckpt_dir_ipa}' , subfolder="", weight_name=["ip_adapter_plus_general.bin"])
+pipe_canny.load_ip_adapter(f'{ckpt_dir_ipa}' , subfolder="", weight_name=["ip_adapter_plus_general.bin"])
 @spaces.GPU
 def process_canny_condition(image, canny_threods=[100,200]):