cdb-boop / ComfyUI-Bringing-Old-Photos-Back-to-Life

Bringing Old Photos Back to Life in ComfyUI.
https://arxiv.org/abs/2004.09484
Other
240 stars 19 forks source link

Question for pth files #10

Open szkane opened 4 days ago

szkane commented 4 days ago
vae_a
Place in models/vae/.

restoration/VAE_A_quality/latest_net_G.pth

vae_b
Extract the following models and place them inside models/vae/.

restoration/VAE_B_quality/latest_net_G.pth
restoration/VAE_B_scratch/latest_net_G.pth (scratch_detection)

which "latest_net_G.pth" mv to the comfyUI path "model/vae" ? All are the some file name, or mv the VAE_XX folder?

cdb-boop commented 4 days ago

which "latest_net_G.pth" mv to the comfyUI path "model/vae"?

If you plan on removing scratches, you need to mv the VAE_B_scratch one. I believe the other one does the restoration without accounting for scratches.

You need to use VAE_A_quality regardless.

All are the some file name, or mv the VAE_XX folder?

You can rename the files or use whatever directory structure you prefer, as they will be selected from a drop-down at runtime.