10 Commits (abfa22c16fb3d9b1ed8d049c7b68e94d1cca5b82)

Author SHA1 Message Date
brkirch abfa22c16f Revert "MPS Upscalers Fix"
This reverts commit 768b95394a8500da639b947508f78296524f1836.
3 years ago
brkirch faed465a0b MPS Upscalers Fix
Get ESRGAN, SCUNet, and SwinIR working correctly on MPS by ensuring memory is contiguous for tensor views before sending to MPS device.
3 years ago
brkirch 4c24347e45 Remove BSRGAN from --use-cpu, add SwinIR 3 years ago
C43H66N12O12S2 ed769977f0 add swinir v2 support 3 years ago
Milly 0609ce06c0 Removed duplicate definition model_path 3 years ago
Alexandre Simard 121ed7d36f Add progress bar for SwinIR in cmd
I do not know how to add them to the UI...
3 years ago
d8ahazard 435fd2112a Fixes, cleanup. 3 years ago
d8ahazard 0dce0df1ee Holy $hit.
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.
3 years ago
d8ahazard 740070ea9c Re-implement universal model loading 3 years ago
d8ahazard bfb7f15d46 Rename swinir -> swinir_model 3 years ago