Spaces:
Paused
Paused
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__":
|