Commit Graph

128 Commits

Author SHA1 Message Date
ZeroCool940711
c35e5804c2 Added a config option to use embeddings from the huggingface stable diffusion concept library. 2022-09-17 15:25:55 -07:00
Chris Heald
27c8134547 Display the final image before running postprocessing, and don't preview when i=0 2022-09-16 18:25:08 -07:00
Chris Heald
d53643822d Collect images for final display
This is a collection of several changes to enhance image display:

* When using GFPGAN or RealESRGAN, only the final output will be
  displayed.
* In batch>1 mode, each final image will be collected into an image grid
  for display
* The image is constrained to a reasonable size to ensure that batch
  grids of RealESRGAN'd images don't end up spitting out a massive image
  that the browser then has to handle.
* Additionally, the progress bar indicator is updated as each image is
  post-processed.
2022-09-16 17:39:46 -07:00
ZeroCool
419f3f9474
Merge branch 'dev' into fix-preview-error 2022-09-16 16:36:55 -07:00
Chris Heald
0cf3d1b0ef When early terminating, generation_callback gets invoked but st.session_state is empty. When this happens, just bail. 2022-09-16 16:29:10 -07:00
ZeroCool
ac99b657d2
Merge pull request #1187 from cheald/enable-batch-preview
When running a batch with preview turned on, produce a grid of previe…
2022-09-16 16:27:43 -07:00
Chris Heald
c1ab63c284 When running a batch with preview turned on, produce a grid of preview images 2022-09-16 16:12:52 -07:00
Chris Heald
d22cf56395
Don't attempt mask restoration when there is no mask given (#1186) 2022-09-16 23:56:09 +03:00
hlky
b856a91ec8
allow webp uploads to img2img tab #991 2022-09-16 20:55:58 +01:00
xaedes
4efe62b31a
add missing argument to GoBig sample function, fixes #1183 (#1184) 2022-09-16 20:40:30 +01:00
ZeroCool940711
ef2da42489 - Bumped the version of diffusers used on the txt2vid tab to be now v0.3.0.
- Added initial file for the textual inversion tab.
2022-09-16 11:50:22 -07:00
Charles Galant
d293073e67 Enable negative prompts on streamlit 2022-09-16 13:25:25 -04:00
ZeroCool
9a6b2173a5
Merge branch 'dev' into gallery-fixes 2022-09-16 04:21:58 -07:00
ZeroCool
e65b8d0b8a
Merge branch 'dev' into fix-find-noise 2022-09-16 04:15:11 -07:00
ZeroCool940711
0645c7cf64 - Increased the max value for the width and height sliders on the txt2img tab.
- Fixed a leftover line from removing the home tab.
2022-09-15 17:18:36 -07:00
ZeroCool940711
88ae534693 Fixed a missing import on the ModelManager script. 2022-09-15 14:27:13 -07:00
ZeroCool940711
eb545c1229 Removed the Home tab until the gallery is fixed. 2022-09-15 14:23:09 -07:00
Chris Heald
51a973929c Fix invalid invocation of find_noise_for_image 2022-09-15 12:48:35 -07:00
Chris Heald
d2f339061b Fix the gallery back/next buttons and add a refresh button 2022-09-15 12:45:45 -07:00
Chris Heald
8a25dcc332 Fix errors rendering galleries when there are not enough images to render 2022-09-15 11:06:56 -07:00
ZeroCool940711
4068d804ad - Added the Home tab made by @devilismyfriend
- Added gallery tab on txt2img.
2022-09-15 06:29:41 -07:00
ZeroCool940711
d93065bb95 Added streamlit_nested_layout component as dependency and imported on the webui_streamli.py file to allow us to use nested columns and expanders. 2022-09-15 03:48:55 -07:00
protoplm
6a19a73388 Fixed update_preview_frequency and update_preview using defaults from
webui_streamlit.yaml instead of state variables from UI.
2022-09-14 23:03:55 -04:00
protoplm
ca6fb326f3 Formatted tabs as spaces 2022-09-14 20:17:07 -04:00
protoplm
4f7adcaf42 Corrected breaking issues introduced in #1136 to txt2img and
made state variables consistent with img2img.

Fixed a bug where switching models after running would not reload
the used model.
2022-09-14 19:59:53 -04:00
ZeroCool940711
3e6a8cc185 Small fix to the custom model list. 2022-09-14 15:18:37 -07:00
ZeroCool940711
02fe63acb2 Added custom models list to the txt2img tab. 2022-09-14 14:48:13 -07:00
cobryan05
81f58d58d0
Re-merge #611 - View/Cancel in-progress diffusions (#796)
* JobManager: Re-merge #611

PR #611 seems to have got lost in the shuffle after
the transition to 'dev'.

This commit re-merges the feature branch. This adds
support for viewing preview images as the image
generates, as well as cancelling in-progress images
and a couple fixes and clean-ups.

* JobManager: Clear jobs that fail to start

Sometimes if a job fails to start it will get stuck in the active job
list. This commit ensures that jobs that raise exceptions are cleared,
and also adds a start timer to clear out jobs that fail to start
within a reasonable amount of time.
2022-09-14 23:48:56 +03:00
ZeroCool940711
3675b566a4 Merge remote-tracking branch 'origin/dev' into dev
# Conflicts:
#	scripts/img2img.py
#	scripts/sd_utils.py
#	scripts/txt2img.py
2022-09-14 11:39:21 -07:00
ZeroCool940711
dbbed38c8c - Fixed max_frame not being properly used and instead sampling_steps was the variable being use.
- Fixed several issues with wrong variable being used on multiple places.
- Addd option to toggle some extra option from the config file for when the model is loading on the txt2vid tab.
2022-09-14 11:34:24 -07:00
ZeroCool940711
4c46633572 Fixed defaults not being loaded correctly or missing in some places. 2022-09-14 09:26:23 -07:00
Thomas Mello
8df0b1a26d temp disable nvml support for multiple gpus 2022-09-14 18:41:42 +03:00
Thomas Mello
ede81bdc5c fix: [streamlit] optimization mode 2022-09-14 18:22:24 +03:00
ZeroCool940711
ed1ebc129c Modified the way the defaults settings are loaded from the config file so we only load them on the webui_strealit.py file and use st.session_state to access them from anywhere else, this makes it so the config can be modified externally like before the code split and the changes will be updated on next rerun of the UI. 2022-09-14 06:40:56 -07:00
ZeroCool
df2c3188c2
Merge branch 'sd-webui:dev' into dev 2022-09-14 06:03:18 -07:00
ZeroCool940711
6a79ea930b Removed condition to check if the defaults are in the st.session_state dictionary, this is not needed and would cause issues with it not being reloaded when the user changes something on it. 2022-09-14 05:58:01 -07:00
unnamedplugins
6ce25a8e0f Merge branch 'dev' of github.com:sd-webui/stable-diffusion-webui into dev 2022-09-14 08:48:21 -04:00
ZeroCool940711
9246692072 Moved defaults to the webui_streamlit.py file and fixed some imports. 2022-09-14 04:19:24 -07:00
ZeroCool940711
1afc89e0d1 Made the defaults settings from the config file be stored inside st.session_state to avoid loading it multiple times when calling the "sd_utils.py" file from other modules. 2022-09-14 03:48:29 -07:00
ZeroCool940711
ef6b39ab6d Fixed the info message on the txt2img tab not showing the info but just showing the text "Done" 2022-09-14 03:20:19 -07:00
ZeroCool940711
d2b3900647 Fixed some imports. 2022-09-14 02:52:02 -07:00
ZeroCool940711
374b144e77 - Added the functions to load the optimized models, this "should" make it so optimized and turbo mode work now but needs to be tested more.
- Added function to load LDSR.
2022-09-14 02:46:01 -07:00
ZeroCool940711
2e684f948f Removed the "find_noise_for_image.py" and "matched_noise.py" scripts as their content is now part of "sd_utils.py" 2022-09-14 02:02:11 -07:00
ZeroCool940711
71be8f09f2 Forgot to call the layout() for the Model Manager tab after the import so it was not been used and the tab was shown as empty. 2022-09-13 23:02:16 -07:00
ZeroCool940711
ede343a269 The webui_streamlit.py file has been split into multiple modules containing their own code making it easier to work with than a single big file.
The list of modules is as follow:
- webuit_streamlit.py: contains the main layout as well as the functions that load the css which is needed by the layout.
- webui_streamlit_old.py: contains the code for the previous version of the WebUI. Will be removed once the new UI code starts to get used and if everything works as it should.
- txt2img.py: contains the code for the txt2img tab.
- img2img.py: contains the code for the img2img tab.
- txt2vid.py: contains the code for the txt2vid tab.
- sd_utils.py: contains utility functions used by more than one module, any function that meets such condition should be placed here.
- ModelManager.py: contains the code for the Model Manager page on the sidebar menu.
- Settings.py: contains the code for the Settings page on the sidebar menu.
- home.py: contains the code for the Home tab, history and gallery implemented by @devilismyfriend.
- imglab.py: contains the code for the Image Lab tab implemented by @devilismyfriend
2022-09-13 14:09:39 -07:00
unnamedplugins
0fc3e592f7 Merge branch 'dev' of github.com:unnamedplugins/stable-diffusion-webui into dev 2022-09-13 10:57:37 -04:00
unnamedplugins
0c0e80d909 Add mask restore feature to streamlit, prevent color correction from modifying initial image when mask_restore is turned on 2022-09-13 10:57:07 -04:00
hlky
92b3159909
half precision streamlit txt2vid
`RuntimeError: expected scalar type Half but found Float` with both `torch_dtype=torch.float16` and `revision="fp16"`
2022-09-13 13:26:36 +01:00
ZeroCool940711
cbbf33d735 Changed the loading of the model on the txt2vid tab so the half models are only loaded if the no_half option on the config file is set to False. 2022-09-13 01:49:03 -07:00
ZeroCool940711
299cef698d Merge remote-tracking branch 'origin/dev' into dev 2022-09-13 00:49:31 -07:00