aboutsummaryrefslogtreecommitdiff
path: root/modules/gfpgan_model.py
AgeCommit message (Collapse)Author
2023-12-31Be more clear about Spandrel model nomenclatureAarni Koskela
2023-12-30Verify architecture for loaded Spandrel modelsAarni Koskela
2023-12-30Unify CodeFormer and GFPGAN restoration backends, use Spandrel for GFPGANAarni Koskela
2023-12-30Use Spandrel for upscaling and face restoration architectures (aside from ↵Aarni Koskela
GFPGAN and LDSR)
2023-10-22fix Blank line contains whitespaceavantcontra
2023-10-22fix bug when using --gfpgan-models-pathavantcontra
2023-06-27Merge pull request #10823 from akx/model-loadyAUTOMATIC1111
Upscaler model loading cleanup
2023-06-13Fix up `if "http" in ...:` to be more sensible startswithsAarni Koskela
2023-06-13Use os.makedirs(..., exist_ok=True)Aarni Koskela
2023-05-31rename print_error to report, use it with together with package nameAUTOMATIC
2023-05-29Add & use modules.errors.print_error where currently printing exception info ↵Aarni Koskela
by hand
2023-05-10F401 fixes for ruffAUTOMATIC
2023-01-27add data-dir flag and set all user data directories based on itMax Audron
2022-11-12Set device for facelib/facexlib and gfpganbrkirch
* 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
2022-10-04send all three of GFPGAN's and codeformer's models to CPU memory instead of ↵AUTOMATIC
just one for #1283
2022-10-04Merge branch 'master' into masterbrkirch
2022-10-03use existing function for gfpganAUTOMATIC
2022-09-30When device is MPS, use CPU for GFPGAN insteadbrkirch
GFPGAN will not work if the device is MPS, so default to CPU instead.
2022-09-30remove unwanted formatting/functionality from the PRAUTOMATIC
2022-09-29Holy $hit.d8ahazard
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
2022-09-23gfpgan: just download the damn modelAUTOMATIC
2022-09-12 Instance of CUDA out of memory on a low-res batch, even with ↵AUTOMATIC
--opt-split-attention-v1 (found cause) #255
2022-09-07codeformer supportAUTOMATIC
2022-09-03option to unload GFPGAN after usingAUTOMATIC
2022-09-03split codebase into multiple files; to anyone this affects negatively: sorryAUTOMATIC