Spaces:
Running
on
Zero
Running
on
Zero
1inkusFace
commited on
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"
|