aboutsummaryrefslogtreecommitdiffstats
path: root/modules/gfpgan_model.py
AgeCommit message (Collapse)AuthorLines
2023-12-31Be more clear about Spandrel model nomenclatureAarni Koskela-4/+6
2023-12-30Verify architecture for loaded Spandrel modelsAarni Koskela-0/+1
2023-12-30Unify CodeFormer and GFPGAN restoration backends, use Spandrel for GFPGANAarni Koskela-112/+54
2023-12-30Use Spandrel for upscaling and face restoration architectures (aside from ↵Aarni Koskela-6/+7
GFPGAN and LDSR)
2023-10-22fix Blank line contains whitespaceavantcontra-1/+1
2023-10-22fix bug when using --gfpgan-models-pathavantcontra-5/+20
2023-06-27Merge pull request #10823 from akx/model-loadyAUTOMATIC1111-1/+1
Upscaler model loading cleanup
2023-06-13Fix up `if "http" in ...:` to be more sensible startswithsAarni Koskela-1/+1
2023-06-13Use os.makedirs(..., exist_ok=True)Aarni Koskela-4/+1
2023-05-31rename print_error to report, use it with together with package nameAUTOMATIC-3/+2
2023-05-29Add & use modules.errors.print_error where currently printing exception info ↵Aarni Koskela-4/+2
by hand
2023-05-10F401 fixes for ruffAUTOMATIC-1/+1
2023-01-27add data-dir flag and set all user data directories based on itMax Audron-3/+2
2022-11-12Set device for facelib/facexlib and gfpganbrkirch-1/+3
* FaceXLib/FaceLib doesn't pass the device argument to RetinaFace but instead chooses one itself and sets it to a global - in order to use a device other than its internally chosen default it is necessary to manually replace the default value * The GFPGAN constructor needs the device argument to work with MPS or a CUDA device ID that differs from the default
2022-10-04Merge branch 'master' into cpu-cmdline-optbrkirch-3/+13
2022-10-04send all three of GFPGAN's and codeformer's models to CPU memory instead of ↵AUTOMATIC-2/+12
just one for #1283
2022-10-04Merge branch 'master' into masterbrkirch-5/+1
2022-10-03use existing function for gfpganAUTOMATIC-5/+1
2022-09-30When device is MPS, use CPU for GFPGAN insteadbrkirch-3/+3
GFPGAN will not work if the device is MPS, so default to CPU instead.
2022-09-30remove unwanted formatting/functionality from the PRAUTOMATIC-9/+3
2022-09-29Holy $hit.d8ahazard-18/+40
Yep. Fix gfpgan_model_arch requirement(s). Add Upscaler base class, move from images. Add a lot of methods to Upscaler. Re-work all the child upscalers to be proper classes. Add BSRGAN scaler. Add ldsr_model_arch class, removing the dependency for another repo that just uses regular latent-diffusion stuff. Add one universal method that will always find and load new upscaler models without having to add new "setup_model" calls. Still need to add command line params, but that could probably be automated. Add a "self.scale" property to all Upscalers so the scalers themselves can do "things" in response to the requested upscaling size. Ensure LDSR doesn't get stuck in a longer loop of "upscale/downscale/upscale" as we try to reach the target upscale size. Add typehints for IDE sanity. PEP-8 improvements. Moar.
2022-09-26Re-implement universal model loadingd8ahazard-31/+29
2022-09-23gfpgan: just download the damn modelAUTOMATIC-6/+13
2022-09-12 Instance of CUDA out of memory on a low-res batch, even with ↵AUTOMATIC-5/+10
--opt-split-attention-v1 (found cause) #255
2022-09-07codeformer supportAUTOMATIC-2/+18
2022-09-03option to unload GFPGAN after usingAUTOMATIC-3/+12
2022-09-03split codebase into multiple files; to anyone this affects negatively: sorryAUTOMATIC-0/+58