aboutsummaryrefslogtreecommitdiffstats
path: root/modules/esrgan_model.py
AgeCommit message (Collapse)AuthorLines
2023-12-30Verify architecture for loaded Spandrel modelsAarni Koskela-0/+1
2023-12-30Use Spandrel for upscaling and face restoration architectures (aside from ↵Aarni Koskela-145/+8
GFPGAN and LDSR)
2023-12-30Refactor esrgan_upscale to more generic upscale_with_modelAarni Koskela-39/+8
2023-06-13Upscaler.load_model: don't return None, just use exceptionsAarni Koskela-8/+6
2023-06-13Add TODO comments to sus model loadsAarni Koskela-0/+1
2023-06-13Fix up `if "http" in ...:` to be more sensible startswithsAarni Koskela-2/+2
2023-06-13Move `load_file_from_url` to modelloaderAarni Koskela-3/+1
2023-05-19change upscalers to download models into user-specified directory (from ↵AUTOMATIC-1/+1
commandline args) rather than the default models/<...>
2023-05-10fixes for B007AUTOMATIC-6/+2
2023-05-10imports cleanup for ruffAUTOMATIC-1/+1
2023-05-09Fix up string formatting/concatenation to f-strings where feasibleAarni Koskela-4/+7
2022-11-17Revert "MPS Upscalers Fix"brkirch-1/+1
This reverts commit 768b95394a8500da639b947508f78296524f1836.
2022-10-30add resrgan 8x, allow use 1x and up to 8x extra models, move BSRGAN model, ↵victorca25-4/+13
add nearest
2022-10-25MPS Upscalers Fixbrkirch-1/+1
Get ESRGAN, SCUNet, and SwinIR working correctly on MPS by ensuring memory is contiguous for tensor views before sending to MPS device.
2022-10-09update ESRGAN architecture and model to support all ESRGAN models in the DB, ↵victorca25-62/+128
BSRGAN and real-ESRGAN models
2022-10-09Removed duplicate definition model_pathMilly-2/+0
2022-10-08Add hypernetwork support to split cross attention v1brkirch-1/+1
* Add hypernetwork support to split_cross_attention_forward_v1 * Fix device check in esrgan_model.py to use devices.device_esrgan instead of shared.device
2022-10-04Add --use-cpu command line optionbrkirch-5/+4
Remove MPS detection to use CPU for GFPGAN / CodeFormer and add a --use-cpu command line option.
2022-10-02Update esrgan_model.pyd8ahazard-2/+2
Use alternate ESRGAN Model download path.
2022-09-30remove unwanted formatting/functionality from the PRAUTOMATIC-61/+62
2022-09-29Holy $hit.d8ahazard-117/+110
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-26Cleanup existing directories, fixesd8ahazard-2/+2
2022-09-26Re-implement universal model loadingd8ahazard-15/+41
2022-09-22Fix/Revert opts name from GAN to ESRGANd8ahazard-2/+2
2022-09-22Add new models, fix shared opts issuesd8ahazard-2/+2
Add General x4x3, GeneralWDN x4x3, and AnimeVideo models from newer ESRGAN releases. Fix issues caused by renaming ESRGAN_tille values to GAN_tile without using an IDE...
2022-09-21Revert "fix for swininr PR breaking ESRGAN for new users"AUTOMATIC-2/+2
This reverts commit 8d1d64f9a238272d7a166d21e25cf529c09b2ad5.
2022-09-21fix for swininr PR breaking ESRGAN for new usersAUTOMATIC-2/+2
2022-09-11Modular device managementAbdullah Barhoum-1/+2
2022-09-08add a helpful message when user puts RealESRGAN model into ESRGAN directory.AUTOMATIC-5/+8
2022-09-07Added support for launching on Apple Siliconxeonvs-2/+5
2022-09-04ESRGAN supportAUTOMATIC-0/+134