Commit Graph

24 Commits

Author SHA1 Message Date
protoplm
e4d82f4e84
Used loaded_model state variable in .yaml generation (#1196)
Used loaded_model state variable in .yaml generation
2022-09-18 10:34:37 +01:00
ZeroCool940711
997eb12733 Basic implementation for the Concept Library tab made by cloning the Home tab. 2022-09-17 19:02:40 -07:00
ZeroCool940711
cd4396d1da Added option to enable enable_attention_slicing and enable_minimal_memory_usage, this for now only works on txt2vid which uses diffusers. 2022-09-17 16:49:42 -07:00
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
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
Charles Galant
d293073e67 Enable negative prompts on streamlit 2022-09-16 13:25:25 -04:00
Chris Heald
51a973929c Fix invalid invocation of find_noise_for_image 2022-09-15 12:48:35 -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
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
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
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
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