Update app.py
Browse files
app.py
CHANGED
|
@@ -118,11 +118,29 @@ def infer(
|
|
| 118 |
enhanced_prompt_2 = prompt
|
| 119 |
if latent_file:
|
| 120 |
sd_image_a = Image.open(latent_file.name).convert('RGB')
|
|
|
|
| 121 |
print("-- using image file and loading ip-adapter --")
|
| 122 |
-
#if image_encoder_path=="laion/CLIP-ViT-H-14-laion2B-s32B-b79K":
|
| 123 |
-
# sd_image_a.resize((384,384), Image.LANCZOS)
|
| 124 |
-
#else:
|
| 125 |
sd_image_a.resize((height,width), Image.LANCZOS)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 126 |
pipe.init_ipadapter(
|
| 127 |
ip_adapter_path=ipadapter_path,
|
| 128 |
image_encoder_path=image_encoder_path,
|
|
@@ -137,8 +155,19 @@ def infer(
|
|
| 137 |
num_inference_steps=num_inference_steps,
|
| 138 |
guidance_scale=guidance_scale,
|
| 139 |
generator=generator,
|
|
|
|
| 140 |
clip_image=sd_image_a,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 141 |
ipadapter_scale=ip_scale,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 142 |
).images[0]
|
| 143 |
timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
|
| 144 |
rv_path = f"sd35_{timestamp}.png"
|
|
|
|
| 118 |
enhanced_prompt_2 = prompt
|
| 119 |
if latent_file:
|
| 120 |
sd_image_a = Image.open(latent_file.name).convert('RGB')
|
| 121 |
+
sd_image_a.resize((height,width), Image.LANCZOS)
|
| 122 |
print("-- using image file and loading ip-adapter --")
|
|
|
|
|
|
|
|
|
|
| 123 |
sd_image_a.resize((height,width), Image.LANCZOS)
|
| 124 |
+
if latent_file_2 is not None: # Check if a latent file is provided
|
| 125 |
+
sd_image_b = Image.open(latent_file_2.name).convert('RGB')
|
| 126 |
+
sd_image_b.resize((height,width), Image.LANCZOS)
|
| 127 |
+
else:
|
| 128 |
+
sd_image_b = None
|
| 129 |
+
if latent_file_3 is not None: # Check if a latent file is provided
|
| 130 |
+
sd_image_c = Image.open(latent_file_3.name).convert('RGB')
|
| 131 |
+
sd_image_c.resize((height,width), Image.LANCZOS)
|
| 132 |
+
else:
|
| 133 |
+
sd_image_c = None
|
| 134 |
+
if latent_file_4 is not None: # Check if a latent file is provided
|
| 135 |
+
sd_image_d = Image.open(latent_file_4.name).convert('RGB')
|
| 136 |
+
sd_image_d.resize((height,width), Image.LANCZOS)
|
| 137 |
+
else:
|
| 138 |
+
sd_image_d = None
|
| 139 |
+
if latent_file_5 is not None: # Check if a latent file is provided
|
| 140 |
+
sd_image_e = Image.open(latent_file_5.name).convert('RGB')
|
| 141 |
+
sd_image_e.resize((height,width), Image.LANCZOS)
|
| 142 |
+
else:
|
| 143 |
+
sd_image_e = None
|
| 144 |
pipe.init_ipadapter(
|
| 145 |
ip_adapter_path=ipadapter_path,
|
| 146 |
image_encoder_path=image_encoder_path,
|
|
|
|
| 155 |
num_inference_steps=num_inference_steps,
|
| 156 |
guidance_scale=guidance_scale,
|
| 157 |
generator=generator,
|
| 158 |
+
max_sequence_length=512,
|
| 159 |
clip_image=sd_image_a,
|
| 160 |
+
clip_image_2=sd_image_b,
|
| 161 |
+
clip_image_3=sd_image_c,
|
| 162 |
+
clip_image_4=sd_image_d,
|
| 163 |
+
clip_image_5=sd_image_e,
|
| 164 |
+
text_scale=text_scale,
|
| 165 |
ipadapter_scale=ip_scale,
|
| 166 |
+
scale_1=latent_file_1_scale,
|
| 167 |
+
scale_2=latent_file_2_scale,
|
| 168 |
+
scale_3=latent_file_3_scale,
|
| 169 |
+
scale_4=latent_file_4_scale,
|
| 170 |
+
scale_5=latent_file_5_scale,
|
| 171 |
).images[0]
|
| 172 |
timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
|
| 173 |
rv_path = f"sd35_{timestamp}.png"
|