Rishi Desai commited on
Commit
847f39e
·
1 Parent(s): 1f03c78

fixes to workflow

Browse files
Files changed (1) hide show
  1. 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
- for q in range(10):
211
- applypulidflux_31 = applypulidflux.apply_pulid_flux(
212
- weight=0.75,
213
- start_at=0.1,
214
- end_at=1,
215
- fusion="mean",
216
- fusion_weight_max=1,
217
- fusion_weight_min=0,
218
- train_step=1000,
219
- use_gray=True,
220
- model=get_value_at_index(unetloader_93, 0),
221
- pulid_flux=get_value_at_index(pulidfluxmodelloader_44, 0),
222
- eva_clip=get_value_at_index(pulidfluxevacliploader_45, 0),
223
- face_analysis=get_value_at_index(pulidfluxinsightfaceloader_46, 0),
224
- image=get_value_at_index(loadimage_24, 0),
225
- unique_id=17601617597548643515,
226
- )
227
-
228
- setunioncontrolnettype_41 = setunioncontrolnettype.set_controlnet_type(
229
- type="tile", control_net=get_value_at_index(controlnetloader_49, 0)
230
- )
231
-
232
- controlnetapplyadvanced_37 = controlnetapplyadvanced.apply_controlnet(
233
- strength=1,
234
- start_percent=0.1,
235
- end_percent=0.8,
236
- positive=get_value_at_index(cliptextencode_42, 0),
237
- negative=get_value_at_index(cliptextencode_23, 0),
238
- control_net=get_value_at_index(setunioncontrolnettype_41, 0),
239
- image=get_value_at_index(loadimage_40, 0),
240
- vae=get_value_at_index(vaeloader_95, 0),
241
- )
242
-
243
- basicguider_122 = basicguider.get_guider(
244
- model=get_value_at_index(applypulidflux_31, 0),
245
- conditioning=get_value_at_index(controlnetapplyadvanced_37, 0),
246
- )
247
-
248
- basicscheduler_131 = basicscheduler.get_sigmas(
249
- scheduler="beta",
250
- steps=28,
251
- denoise=0.75,
252
- model=get_value_at_index(applypulidflux_31, 0),
253
- )
254
-
255
- samplercustomadvanced_1 = samplercustomadvanced.sample(
256
- noise=get_value_at_index(randomnoise_39, 0),
257
- guider=get_value_at_index(basicguider_122, 0),
258
- sampler=get_value_at_index(ksamplerselect_50, 0),
259
- sigmas=get_value_at_index(basicscheduler_131, 0),
260
- latent_image=get_value_at_index(vaeencode_35, 0),
261
- )
262
-
263
- vaedecode_114 = vaedecode.decode(
264
- samples=get_value_at_index(samplercustomadvanced_1, 0),
265
- vae=get_value_at_index(vaeloader_95, 0),
266
- )
267
-
268
- faceembeddistance_117 = faceembeddistance.analize(
269
- similarity_metric="cosine",
270
- filter_thresh=100,
271
- filter_best=0,
272
- generate_image_overlay=True,
273
- analysis_models=get_value_at_index(faceanalysismodels_118, 0),
274
- reference=get_value_at_index(loadimage_24, 0),
275
- image=get_value_at_index(vaedecode_114, 0),
276
- )
277
-
278
- display_any_rgthree_121 = display_any_rgthree.main(
279
- source=get_value_at_index(faceembeddistance_117, 1)
280
- )
281
-
282
- image_comparer_rgthree_123 = image_comparer_rgthree.compare_images(
283
- image_a=get_value_at_index(loadimage_40, 0),
284
- image_b=get_value_at_index(vaedecode_114, 0),
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__":