Spaces:
Running
on
Zero
Running
on
Zero
Rishi Desai
commited on
Commit
·
847f39e
1
Parent(s):
1f03c78
fixes to workflow
Browse files- FaceEnhancementProdApi.py +76 -88
FaceEnhancementProdApi.py
CHANGED
@@ -203,96 +203,84 @@ def main():
|
|
203 |
samplercustomadvanced = NODE_CLASS_MAPPINGS["SamplerCustomAdvanced"]()
|
204 |
vaedecode = VAEDecode()
|
205 |
faceembeddistance = NODE_CLASS_MAPPINGS["FaceEmbedDistance"]()
|
206 |
-
display_any_rgthree = NODE_CLASS_MAPPINGS["Display Any (rgthree)"]()
|
207 |
-
image_comparer_rgthree = NODE_CLASS_MAPPINGS["Image Comparer (rgthree)"]()
|
208 |
saveimage = SaveImage()
|
209 |
|
210 |
-
|
211 |
-
|
212 |
-
|
213 |
-
|
214 |
-
|
215 |
-
|
216 |
-
|
217 |
-
|
218 |
-
|
219 |
-
|
220 |
-
|
221 |
-
|
222 |
-
|
223 |
-
|
224 |
-
|
225 |
-
|
226 |
-
|
227 |
-
|
228 |
-
|
229 |
-
|
230 |
-
|
231 |
-
|
232 |
-
|
233 |
-
|
234 |
-
|
235 |
-
|
236 |
-
|
237 |
-
|
238 |
-
|
239 |
-
|
240 |
-
|
241 |
-
|
242 |
-
|
243 |
-
|
244 |
-
|
245 |
-
|
246 |
-
|
247 |
-
|
248 |
-
|
249 |
-
|
250 |
-
|
251 |
-
|
252 |
-
|
253 |
-
|
254 |
-
|
255 |
-
|
256 |
-
|
257 |
-
|
258 |
-
|
259 |
-
|
260 |
-
|
261 |
-
|
262 |
-
|
263 |
-
|
264 |
-
|
265 |
-
|
266 |
-
|
267 |
-
|
268 |
-
|
269 |
-
|
270 |
-
|
271 |
-
|
272 |
-
|
273 |
-
|
274 |
-
|
275 |
-
|
276 |
-
|
277 |
-
|
278 |
-
|
279 |
-
|
280 |
-
|
281 |
-
|
282 |
-
|
283 |
-
|
284 |
-
|
285 |
-
|
286 |
-
|
287 |
-
saveimage_128 = saveimage.save_images(
|
288 |
-
filename_prefix="FaceEnhanced",
|
289 |
-
images=get_value_at_index(vaedecode_114, 0),
|
290 |
-
)
|
291 |
-
|
292 |
-
saveimage_129 = saveimage.save_images(
|
293 |
-
filename_prefix="FaceEmbedDist",
|
294 |
-
images=get_value_at_index(faceembeddistance_117, 0),
|
295 |
-
)
|
296 |
|
297 |
|
298 |
if __name__ == "__main__":
|
|
|
203 |
samplercustomadvanced = NODE_CLASS_MAPPINGS["SamplerCustomAdvanced"]()
|
204 |
vaedecode = VAEDecode()
|
205 |
faceembeddistance = NODE_CLASS_MAPPINGS["FaceEmbedDistance"]()
|
|
|
|
|
206 |
saveimage = SaveImage()
|
207 |
|
208 |
+
applypulidflux_31 = applypulidflux.apply_pulid_flux(
|
209 |
+
weight=0.75,
|
210 |
+
start_at=0.1,
|
211 |
+
end_at=1,
|
212 |
+
fusion="mean",
|
213 |
+
fusion_weight_max=1,
|
214 |
+
fusion_weight_min=0,
|
215 |
+
train_step=1000,
|
216 |
+
use_gray=True,
|
217 |
+
model=get_value_at_index(unetloader_93, 0),
|
218 |
+
pulid_flux=get_value_at_index(pulidfluxmodelloader_44, 0),
|
219 |
+
eva_clip=get_value_at_index(pulidfluxevacliploader_45, 0),
|
220 |
+
face_analysis=get_value_at_index(pulidfluxinsightfaceloader_46, 0),
|
221 |
+
image=get_value_at_index(loadimage_24, 0),
|
222 |
+
unique_id=17601617597548643515,
|
223 |
+
)
|
224 |
+
|
225 |
+
setunioncontrolnettype_41 = setunioncontrolnettype.set_controlnet_type(
|
226 |
+
type="tile", control_net=get_value_at_index(controlnetloader_49, 0)
|
227 |
+
)
|
228 |
+
|
229 |
+
controlnetapplyadvanced_37 = controlnetapplyadvanced.apply_controlnet(
|
230 |
+
strength=1,
|
231 |
+
start_percent=0.1,
|
232 |
+
end_percent=0.8,
|
233 |
+
positive=get_value_at_index(cliptextencode_42, 0),
|
234 |
+
negative=get_value_at_index(cliptextencode_23, 0),
|
235 |
+
control_net=get_value_at_index(setunioncontrolnettype_41, 0),
|
236 |
+
image=get_value_at_index(loadimage_40, 0),
|
237 |
+
vae=get_value_at_index(vaeloader_95, 0),
|
238 |
+
)
|
239 |
+
|
240 |
+
basicguider_122 = basicguider.get_guider(
|
241 |
+
model=get_value_at_index(applypulidflux_31, 0),
|
242 |
+
conditioning=get_value_at_index(controlnetapplyadvanced_37, 0),
|
243 |
+
)
|
244 |
+
|
245 |
+
basicscheduler_131 = basicscheduler.get_sigmas(
|
246 |
+
scheduler="beta",
|
247 |
+
steps=28,
|
248 |
+
denoise=0.75,
|
249 |
+
model=get_value_at_index(applypulidflux_31, 0),
|
250 |
+
)
|
251 |
+
|
252 |
+
samplercustomadvanced_1 = samplercustomadvanced.sample(
|
253 |
+
noise=get_value_at_index(randomnoise_39, 0),
|
254 |
+
guider=get_value_at_index(basicguider_122, 0),
|
255 |
+
sampler=get_value_at_index(ksamplerselect_50, 0),
|
256 |
+
sigmas=get_value_at_index(basicscheduler_131, 0),
|
257 |
+
latent_image=get_value_at_index(vaeencode_35, 0),
|
258 |
+
)
|
259 |
+
|
260 |
+
vaedecode_114 = vaedecode.decode(
|
261 |
+
samples=get_value_at_index(samplercustomadvanced_1, 0),
|
262 |
+
vae=get_value_at_index(vaeloader_95, 0),
|
263 |
+
)
|
264 |
+
|
265 |
+
faceembeddistance_117 = faceembeddistance.analize(
|
266 |
+
similarity_metric="cosine",
|
267 |
+
filter_thresh=100,
|
268 |
+
filter_best=0,
|
269 |
+
generate_image_overlay=True,
|
270 |
+
analysis_models=get_value_at_index(faceanalysismodels_118, 0),
|
271 |
+
reference=get_value_at_index(loadimage_24, 0),
|
272 |
+
image=get_value_at_index(vaedecode_114, 0),
|
273 |
+
)
|
274 |
+
|
275 |
+
saveimage_128 = saveimage.save_images(
|
276 |
+
filename_prefix="FaceEnhanced",
|
277 |
+
images=get_value_at_index(vaedecode_114, 0),
|
278 |
+
)
|
279 |
+
|
280 |
+
saveimage_129 = saveimage.save_images(
|
281 |
+
filename_prefix="FaceEmbedDist",
|
282 |
+
images=get_value_at_index(faceembeddistance_117, 0),
|
283 |
+
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
284 |
|
285 |
|
286 |
if __name__ == "__main__":
|