2cacbc124c
load_spandrel_model: make `half` `prefer_half`
...
As discussed with the Spandrel folks, it's good to heed Spandrel's
"supports half precision" flag to avoid e.g. black blotches and what-not.
2024-01-02 10:44:38 +02:00
777af661a2
Be more clear about Spandrel model nomenclature
2023-12-31 00:22:58 +02:00
393a5b82ba
Correct RealESRGAN expected architecture type to ESRGAN
2023-12-30 21:12:32 +02:00
4ad0c0c0a8
Verify architecture for loaded Spandrel models
2023-12-30 16:37:03 +02:00
b0f5934234
Use Spandrel for upscaling and face restoration architectures (aside from GFPGAN and LDSR)
2023-12-30 16:24:01 +02:00
f29b4cd7cb
tell RealESRGANer which device to run on, could be cuda, M1, or other GPU
2023-08-23 14:31:38 +08:00
bf67a5dcf4
Upscaler.load_model: don't return None, just use exceptions
2023-06-13 12:44:25 +03:00
89352a2f52
Move `load_file_from_url` to modelloader
2023-06-13 12:38:28 +03:00
05933840f0
rename print_error to report, use it with together with package name
2023-05-31 19:56:37 +03:00
00dfe27f59
Add & use modules.errors.print_error where currently printing exception info by hand
2023-05-29 09:17:30 +03:00
df6fffb054
change upscalers to download models into user-specified directory (from commandline args) rather than the default models/<...>
2023-05-19 09:09:18 +03:00
4b854806d9
F401 fixes for ruff
2023-05-10 09:02:23 +03:00
762265eab5
autofixes from ruff
2023-05-10 07:52:45 +03:00
3ba6c3c83c
Fix up string formatting/concatenation to f-strings where feasible
2023-05-09 22:25:39 +03:00
539a69860b
fix `--realesrgan-models-path` not working
2023-04-04 20:39:51 +08:00
ada17dbd7c
Refactor conditional casting, fix upscalers
2023-01-28 04:16:25 -05:00
aede265f1d
Fix unable to find Real-ESRGAN model info error (AttributeError: 'NoneType' object has no attribute 'data_path') #6841 #5170
2023-01-17 13:57:55 +03:00
0609ce06c0
Removed duplicate definition model_path
2022-10-09 12:46:07 +03:00
d1f098540a
remove unwanted formatting/functionality from the PR
2022-09-30 11:42:40 +03:00
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.
2022-09-29 17:46:23 -05:00
740070ea9c
Re-implement universal model loading
2022-09-26 09:29:50 -05:00
3eaae8ae65
reworking #775
2022-09-22 07:59:27 +03:00
c214c428bc
Basic settings for realesrgan model selection.
...
I don't like that you have to restart the app, but it works.
2022-09-22 07:21:54 +03:00
4f434c8aea
Fix/Revert opts name from GAN to ESRGAN
2022-09-22 07:21:54 +03:00
9a93a82b82
Cleanup Imports
2022-09-22 07:21:54 +03:00
3cc68300cb
Typo Fix
2022-09-22 07:21:54 +03:00
dd5566814a
Add new models, fix shared opts issues
...
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-22 07:21:54 +03:00
6785fabefb
Revert "fix for swininr PR breaking ESRGAN for new users"
...
This reverts commit 8d1d64f9a2
2022-09-21 09:09:20 +03:00
8d1d64f9a2
fix for swininr PR breaking ESRGAN for new users
2022-09-21 08:39:47 +03:00
f5001246e2
honor tiling settings for RealESRGAN also
...
load scripts earlier to get errors before model loads
2022-09-08 15:19:36 +03:00
f299645aee
ESRGAN support
2022-09-04 18:54:12 +03:00
345028099d
split codebase into multiple files; to anyone this affects negatively: sorry
2022-09-03 12:08:45 +03:00
Aarni Koskela