aboutsummaryrefslogtreecommitdiff
path: root/modules/esrgan_model.py
AgeCommit message (Collapse)Author
2023-05-19change upscalers to download models into user-specified directory (from ↵AUTOMATIC
commandline args) rather than the default models/<...>
2023-05-10fixes for B007AUTOMATIC
2023-05-10imports cleanup for ruffAUTOMATIC
2023-05-09Fix up string formatting/concatenation to f-strings where feasibleAarni Koskela
2022-11-17Revert "MPS Upscalers Fix"brkirch
This reverts commit 768b95394a8500da639b947508f78296524f1836.
2022-10-30add resrgan 8x, allow use 1x and up to 8x extra models, move BSRGAN model, ↵victorca25
add nearest
2022-10-25MPS Upscalers Fixbrkirch
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
BSRGAN and real-ESRGAN models
2022-10-09Removed duplicate definition model_pathMilly
2022-10-08Add hypernetwork support to split cross attention v1brkirch
* 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
Remove MPS detection to use CPU for GFPGAN / CodeFormer and add a --use-cpu command line option.
2022-10-02Update esrgan_model.pyd8ahazard
Use alternate ESRGAN Model download path.
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-26Cleanup existing directories, fixesd8ahazard
2022-09-26Re-implement universal model loadingd8ahazard
2022-09-22Fix/Revert opts name from GAN to ESRGANd8ahazard
2022-09-22Add new models, fix shared opts issuesd8ahazard
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
This reverts commit 8d1d64f9a238272d7a166d21e25cf529c09b2ad5.
2022-09-21fix for swininr PR breaking ESRGAN for new usersAUTOMATIC
2022-09-11Modular device managementAbdullah Barhoum
2022-09-08add a helpful message when user puts RealESRGAN model into ESRGAN directory.AUTOMATIC
2022-09-07Added support for launching on Apple Siliconxeonvs
2022-09-04ESRGAN supportAUTOMATIC