aboutsummaryrefslogtreecommitdiff
path: root/modules/modelloader.py
AgeCommit message (Collapse)Author
2023-02-04Update modelloader.pyspezialspezial
os.path.getmtime(filename) throws exception later in codepath when meeting broken symlink. For now catch it here early but more checks could be added for robustness.
2023-01-10add more stuff to ignore when creating model from configAUTOMATIC
prevent .vae.safetensors files from being listed as stable diffusion models
2023-01-03Make it so that upscalers are not repeated when restarting UI.AUTOMATIC
2022-12-03add built-in extension systemAUTOMATIC
add support for adding upscalers in extensions move LDSR, ScuNET and SwinIR to built-in extensions
2022-11-19Label and load SD .safetensors model filesTim Patton
2022-10-30add resrgan 8x, allow use 1x and up to 8x extra models, move BSRGAN model, ↵victorca25
add nearest
2022-10-02Merge branch 'master' into ScuNETd8ahazard
2022-10-02fix incorrect use of glob in modelloader for #1410AUTOMATIC
2022-09-30Optimize model loaderd8ahazard
Child classes only get populated to __subclassess__ when they are imported. We don't actually need to import any of them to webui any more, so clean up webUI imports and make sure loader imports children. Also, fix command line paths not actually being passed to the scalers.
2022-09-30Logging....d8ahazard
Cleanup. Sorry.
2022-09-30Fix recursive model loadingd8ahazard
Ensure we find checkpoints within subdirectories.
2022-09-30remove unwanted formatting/functionality from the PRAUTOMATIC
2022-09-30undo upscaler model filename tamperingAUTOMATIC
2022-09-29Fixes, cleanup.d8ahazard
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-27Remove debugging messaged8ahazard
2022-09-27Use model loader with stable-diffusion too.d8ahazard
Hook the model loader into the SD_models file. Add default url/download if checkpoint is not found. Add matching stablediffusion-models-path argument. Add message that --ckpt-dir will be removed in the future, but have it pipe to stablediffusion-models-path for now. Update help strings for models-path args so they're more or less uniform. Move sd_model "setup" call to webUI with the others. Ensure "cleanup_models" method moves existing models to the new locations, including SD, and that we aren't deleting folders that still have stuff in them.
2022-09-26Cleanup existing directories, fixesd8ahazard
2022-09-26Re-implement universal model loadingd8ahazard