1inkusFace commited on
Commit
70ebf3d
·
verified ·
1 Parent(s): 4f376ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +32 -3
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"