Commit Graph

220 Commits

Author SHA1 Message Date
discus0434 6021f7a75f add options to custom hypernetwork layer structure 2022-10-19 00:51:36 +09:00
arcticfaded 8d5d863a9d gradio and FastAPI 2022-10-18 06:51:53 +00:00
Ryan Voots 1df3ff25e6 Add --nowebui as a means of disabling the webui and run on the other port 2022-10-18 08:44:50 +03:00
Ryan Voots c3851a853d Re-use webui fastapi application rather than requiring one or the other, not both. 2022-10-18 08:44:50 +03:00
AUTOMATIC d42125baf6 add missing requirement for api and fix some typos 2022-10-17 11:50:20 +03:00
AUTOMATIC 964b63c042 add api() function to return webui() to how it was 2022-10-17 11:38:32 +03:00
Jonathan f3fe487e63 Update webui.py 2022-10-17 11:34:22 +03:00
arcticfaded 60251c9456 initial prototype by borrowing contracts 2022-10-17 11:34:22 +03:00
AUTOMATIC 354ef0da3b add hypernetwork multipliers 2022-10-13 20:12:37 +03:00
AUTOMATIC 8aead63f1a emergency fix 2022-10-12 09:32:14 +03:00
AUTOMATIC fd07b103ae prevent SD model from loading when running in deepdanbooru process 2022-10-12 09:00:39 +03:00
AUTOMATIC1111 f7e86aa420
Merge pull request #2227 from papuSpartan/master
Refresh list of models/ckpts upon hitting restart gradio in the setti…
2022-10-11 17:15:19 +03:00
papuSpartan d01a2d0156 move list refresh to webui.py and add stdout indicating it's doing so 2022-10-11 08:31:28 -05:00
AUTOMATIC b0583be088 more renames 2022-10-11 15:54:34 +03:00
AUTOMATIC 873efeed49 rename hypernetwork dir to hypernetworks to prevent clash with an old filename that people who use zip instead of git clone will have 2022-10-11 15:51:30 +03:00
AUTOMATIC 530103b586 fixes related to merge 2022-10-11 14:53:02 +03:00
AUTOMATIC 5de806184f Merge branch 'master' into hypernetwork-training 2022-10-11 11:14:36 +03:00
Fampai 122d42687b Fix VRAM Issue by only loading in hypernetwork when selected in settings 2022-10-09 11:08:11 +03:00
DepFA 34acad1628 Add GZipMiddleware to root demo 2022-10-08 18:03:16 +03:00
Trung Ngo 786d9f63aa Add button to skip the current iteration 2022-10-08 13:40:39 +03:00
AUTOMATIC 12c4d5c6b5 hypernetwork training mk1 2022-10-07 23:22:22 +03:00
AUTOMATIC 2a53280495 reorder imports to fix the bug with k-diffusion on some version 2022-10-06 13:21:32 +03:00
AUTOMATIC 5f24b7bcf4 option to let users select which samplers they want to hide 2022-10-06 12:08:59 +03:00
AUTOMATIC c0389eb307 hello 2022-10-02 21:35:29 +03:00
AUTOMATIC1111 688c4a914a
Merge branch 'master' into 1404-script-reload-without-restart 2022-10-02 21:33:22 +03:00
DepFA a634c3226f
correct indent 2022-10-02 19:26:38 +01:00
AUTOMATIC a1cde7e646 disabled SD model download after multiple complaints 2022-10-02 21:09:10 +03:00
d8ahazard 5d26ba2b4b
Merge branch 'master' into ScuNET 2022-10-02 07:56:22 -05:00
AUTOMATIC 820f1dc96b initial support for training textual inversion 2022-10-02 15:03:39 +03:00
DepFA 30f2e35658
add importlib.reload 2022-10-01 22:50:03 +01:00
DepFA a9044475c0
add time import 2022-10-01 21:47:42 +01:00
DepFA 95f35d04ab
Host busy thread, check for reload 2022-10-01 18:31:58 +01:00
d8ahazard e82ea20299 Optimize model loader
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-30 15:26:18 -05:00
AUTOMATIC d1f098540a remove unwanted formatting/functionality from the PR 2022-09-30 11:42:40 +03:00
d8ahazard d73741794d Merge remote-tracking branch 'upstream/master' into ModelLoader 2022-09-29 19:59:36 -05:00
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.
2022-09-29 17:46:23 -05:00
AUTOMATIC 2f2d356e4c call torch_gc before/after each gpu gradio operation 2022-09-29 11:32:12 +03:00
d8ahazard 5756d517a6 Merge remote-tracking branch 'upstream/master' into ModelLoader 2022-09-27 11:02:41 -05:00
d8ahazard 11875f5863 Use model loader with stable-diffusion too.
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-27 11:01:13 -05:00
d8ahazard 7d5c29b674 Cleanup existing directories, fixes 2022-09-26 10:27:18 -05:00
d8ahazard 740070ea9c Re-implement universal model loading 2022-09-26 09:29:50 -05:00
William Moorehouse 91643f651d Add support for checkpoint merging 2022-09-25 19:22:12 -04:00
Eyrie 1877a3767e Changed job_id to timestamp 2022-09-25 17:11:07 +03:00
Eyrie 40166dbf08 Added job_id pattern for directories 2022-09-25 17:11:07 +03:00
AUTOMATIC c82e32652e prevent weird behavior when pressing interrupt just as image finishes 2022-09-23 08:48:19 +03:00
d8ahazard 1a1f7e85c7
Add LDSR and "GoLatent?" Upscaling (#763)
* Add LDSR Upscaling
2022-09-21 16:06:37 +03:00
C43H66N12O12S2 5f71ecfe6f Update webui.py 2022-09-20 23:31:06 +03:00
C43H66N12O12S2 62ca23ac5b add swinir import 2022-09-20 23:31:06 +03:00
C43H66N12O12S2 ae8e21abdd hack 2022-09-20 23:31:06 +03:00
Connum f3965cce01 added command line argument --autolaunch to open the webui URL in the system's default browser upon launch (implements #274) 2022-09-18 22:39:34 +03:00
AUTOMATIC 247f58a5e7 add support for switching model checkpoints at runtime 2022-09-17 12:05:18 +03:00
AUTOMATIC 3de44fc580 Include the model name (or the SHA256 of the file) in the metadata #271 2022-09-12 20:47:46 +03:00
AUTOMATIC 834b6e396b aaaaaaaaaaaaaaaaa 2022-09-12 16:52:06 +03:00
AUTOMATIC 89d94e13a7 rename --channelslast to --opt-channelslast to be in line with other torch optimizations 2022-09-12 16:51:23 +03:00
C43H66N12O12S2 fbeadef130
webui.py channels last support 2022-09-12 16:26:42 +03:00
AUTOMATIC 2938dc39fc fixed gradio auth bug in PR 2022-09-12 15:52:16 +03:00
EyeDeck fc49844aa8
Merge branch 'master' into master 2022-09-12 08:19:14 -04:00
AUTOMATIC 372a2c3e2e [Feature Request] Please add "--share-password" for Gradio server security #315 2022-09-12 15:10:05 +03:00
EyeDeck e3646e79aa Add --auth command line argument to enable Gradio authentication
Allows you to pass in Gradio authentication like:
`--auth username:password`
Supports multiple sets of credentials by comma-delimiting, like:
`--auth user1:pass1,user2:pass3`...
2022-09-12 08:08:41 -04:00
AUTOMATIC ddc86f2edb --gradio-debug for collab users 2022-09-12 12:40:55 +03:00
AUTOMATIC f194457229 CLIP interrogator 2022-09-11 18:48:36 +03:00
AUTOMATIC f185874418 [Feature Request] Save defaults for extras & keep image parameters after using extras #251 2022-09-11 11:31:16 +03:00
AUTOMATIC 02bcd51a5a fix aggressive caching for extras tab 2022-09-08 23:29:36 +03:00
AUTOMATIC f5001246e2 honor tiling settings for RealESRGAN also
load scripts earlier to get errors before model loads
2022-09-08 15:19:36 +03:00
AUTOMATIC 3a4c6d9ef5 add webui() function for more simple cell in the notebook 2022-09-08 12:17:26 +03:00
orionaskatu cce6f1df41
fix default 2022-09-08 10:46:23 +02:00
orionaskatu 48317a5176
Port defaults to none 2022-09-08 10:40:56 +02:00
orionaskatu 4f3cebd51d
Add server_port param to webui.py 2022-09-08 09:46:28 +02:00
AUTOMATIC 8dcd3e57e7 a little bit of rework for extras tab 2022-09-07 13:35:02 +03:00
AUTOMATIC 6a9b33c848 codeformer support 2022-09-07 12:32:28 +03:00
AUTOMATIC 7ce7fb01e0 fix for live progress breaking lowvram and medvram optimizations 2022-09-06 23:10:12 +03:00
AUTOMATIC 7e5b8becf9 fix for progress bar 2022-09-06 20:26:09 +03:00
AUTOMATIC fd66199769 added preview option 2022-09-06 19:33:51 +03:00
AUTOMATIC 395f170597 readme extras for VRAM for
added missing packages to requirements for #74
add support for negative numbers in X/Y plot (plus ranges) #73
changed progressbar to work properly with custom modes
2022-09-06 10:11:25 +03:00
AUTOMATIC beece7d85c allow import as library for #72 2022-09-06 08:54:11 +03:00
AUTOMATIC a243bc7859 added progressbar
added an option to disable progressbar
added interrupt support to DDIM/PLMS
2022-09-06 02:09:01 +03:00
AUTOMATIC 064965c466 added --listen option to make gradio listen on network requests 2022-09-05 19:37:11 +03:00
AUTOMATIC a8a58dbac7 re-integrated tiling option as a UI element 2022-09-05 03:25:37 +03:00
AUTOMATIC f91d0c3d19 add an option to enable tiling image generation 2022-09-05 02:16:36 +03:00
AUTOMATIC f299645aee ESRGAN support 2022-09-04 18:54:12 +03:00
hentailord85ez e23a36b666
Fixed typo 'medram' -> 'medvram' in webui.py 2022-09-03 16:47:37 +01:00
AUTOMATIC f40617d6c4 support for scripts 2022-09-03 17:21:15 +03:00
AUTOMATIC 345028099d split codebase into multiple files; to anyone this affects negatively: sorry 2022-09-03 12:08:45 +03:00
AUTOMATIC d7b67d9b40 load GFPGAN once and keep it in memory for #53 2022-09-03 09:05:48 +03:00
AUTOMATIC 0dca0db7eb Update to support embedding with length greater than 1. 2022-09-03 01:01:58 +03:00
AUTOMATIC 4cafad66d2 this should send the currently selected image to other tabs instead of the first 2022-09-02 23:25:29 +03:00
AUTOMATIC efa0a6483c change webui.bat to use venv 2022-09-02 19:26:18 +03:00
AUTOMATIC 2000830342 Change SD upscales to save as images rather than grids and to respect image saving settings. 2022-09-02 15:33:01 +03:00
deggua 75fde6bcfe
Merge branch 'master' into automatic_master 2022-09-02 07:44:19 -04:00
deggua 4e2cc78e52 Removed unnecessary condition check for masking 2022-09-02 07:41:10 -04:00
AUTOMATIC 7ec7a3b121 fix for bad data in saved text files when batch size or batch cunt > 1 2022-09-02 14:33:43 +03:00
deggua 422981c85b Forgot to remove the broken lines 2022-09-02 07:28:39 -04:00
deggua f9c0046897 Fix for fill latent zeros/noise in masking mode 2022-09-02 07:18:12 -04:00
AUTOMATIC 38a483f976 buttons for sending images to tabs 2022-09-02 12:04:36 +03:00
AUTOMATIC 67216187a8 bat file for installing and launching 2022-09-02 09:49:35 +03:00
AUTOMATIC 8f7802f6d8 saving text files along with images for >>88400870 2022-09-02 02:12:50 +03:00
AUTOMATIC 6970696789 fix for torch 1.12.1 loading saved file from torch 1.11 2022-09-02 01:31:01 +03:00
AUTOMATIC d5dfbc45a6 added detection if SD repo is located in a current directory 2022-09-02 01:06:11 +03:00
AUTOMATIC b90feb02aa option to disable ((())) and [[[]]] 2022-09-02 00:30:49 +03:00
AUTOMATIC 0a85f1afff fix error messages not appearing 2022-09-02 00:24:30 +03:00