Commit Graph

48 Commits

Author SHA1 Message Date
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
ZeroCool940711
9246692072 Moved defaults to the webui_streamlit.py file and fixed some imports. 2022-09-14 04:19:24 -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
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
ZeroCool940711
67cb1116c8 Made it so the sampling_steps is added to num_inference_steps, otherwise it would not match the value you set for it on the slider. 2022-09-13 00:47:30 -07:00
ZeroCool
0480804499
Merge branch 'dev' into feature/img2img-masking-fixes 2022-09-13 00:06:34 -07:00
Brian Semrau
003f97d52a img2img mask fixes and fix image2noise normalization 2022-09-13 02:04:26 -04:00
unnamedplugins
2af42fed51 Fix indentation error that prevents mask_restore from working unless ESRGAN is turned on 2022-09-12 23:01:59 -04:00
unnamedplugins
acc5f7591a Fix issue where loopback was using batch mode 2022-09-12 21:43:53 -04:00
ZeroCool940711
f80a2f69ca Commented some print statements that were used for debugging and forgot to remove previously. 2022-09-12 12:01:16 -07:00
ZeroCool940711
1117b5604a -Added the Update Image Preview option to be part of the current tab options under Preview Settings.
- Added Dynamic Preview Frequency option for the txt2vid tab which tries to find the lowest value for update_preview_frequency at which we can update the preview image during generation while at the same time minimizing the impact it has in performance.
- Added option to save a video file on the outputs/txt2vid-samples folder after the generation is complete similar to how the save_grid option works on other tabs.
- Added a video preview which shows a video on the txt2vid tab when the generation is completed.
- Formated some lines of code to make it use less space and fit on the a single screen.
- Added a script called Settings.py to the script folder in which Settings for the Setting page will be placed. Empty for now.
2022-09-12 11:44:00 -07:00
ZeroCool
ba63fd7004
Merge pull request #1040 from briansemrau/feature/st-img2img-masking
Add masking to streamlit img2img
2022-09-12 00:24:59 -07:00
Brian Semrau
1089ec1162 add masking to streamlit img2img, find_noise_for_image, matched_noise 2022-09-12 03:02:43 -04:00
ZeroCool940711
9c8ac62b3d Fixed an error with the metadata not able to be saved because of the seed was not converted to a string before so it had no attribute encode on it. 2022-09-11 23:57:29 -07:00
ZeroCool940711
36a1551465 - Added code for @Hafiidz's changes made on the css. 2022-09-11 23:11:42 -07:00
ZeroCool940711
20172d381e Multiple improvements to the txt2vid tab.
- Improved txt2vid speed by 2 times.
- Added DDIM scheduler.
- Added sliders for beta_start and beta_end to have more control over these parameters on the scheduler.
- Added option to select the scheduler type from scaled_linear or linear.
- Added option to save info files for the txt2vid tab and improved the information saved to include most of the parameters used to run the generation.
- You can now download any model from the huggingface website to use on the txt2vid tab, just add the name to the custom_models_list on the config file.
2022-09-11 10:42:17 -07:00
ZeroCool940711
e6463e3249 Added some basic layout for the Model Manager tab and added there the models that most people use to make it easy to download instead of having to go do the wiki or searching through discord for links, it also shows the path where you are supposed to put those models for them to work. 2022-09-11 03:35:14 -07:00
JJ
42cc7ea8dc fix individual image file format saves
* introduces a general config setting save_format similar to grid_format for individual file saves
2022-09-11 17:40:28 +10:00
JJ
2672743f67 write same metadata to file and yaml
* include piexif in environment needed for exif labelling of non-png files
2022-09-11 17:27:52 +10:00
ZeroCool
a760e036d9
Commented an import that is not used for now but will be used soon. 2022-09-10 21:31:42 -07:00
ZeroCool940711
fa08f00ae6 Changed the minimum value for the Sampling Steps and Inference Steps to 10 and added step with a value of 10 to make it easier to move the slider as it will require a higher maximum value than in other tabs for good results on the text2vid tab. 2022-09-10 21:24:22 -07:00
ZeroCool
be2ece0683
Merge pull request #959 from jjisnow/new_metadata
New metadata for other image types
2022-09-10 19:11:32 -07:00
JJ
8e7959ca2b add metadata to other file output file types 2022-09-11 09:49:14 +10:00
JJ
b4db9733d0 add sampler_name defaults to img2img 2022-09-11 09:37:14 +10:00
ZeroCool940711
80b00810ef - Improved the progress bar for the txt2vid tab, it now shows more information during generation.
- Changed the guidance_scale variable to be cfg_scale.
2022-09-10 07:42:20 -07:00
ZeroCool940711
2795dfb8ab Changed the scheduler for the txt2vid tab back to LMS, for now we can only use that. 2022-09-09 19:51:28 -07:00
ZeroCool940711
249e2dd74f Fixed an issue with the txt2vid model. 2022-09-09 19:43:51 -07:00
xaedes
f6aa2c64eb
Add mask_restore to restore images based on mask, fixing #665 (#898)
* Add mask_restore option to give users the option to restore images based on mask, fixing #665.

Before commit c73fdd78  (Implement masking during sampling to improve blending, #308)
image mask was applied after sampling, resulting in masked parts that are not regenerated
to actually stay the same.
Since c73fdd78 the masked img2img will change the whole image, even in masked areas.
It gives better looking results at first glance, but will result in image degredation when
applied a few times. See issue #665.

In the workflow of using repeated masked img2img, users may want to use this options to keep the parts
of image they actually want to keep without image degradation. A final masked img2img or whole image img2img with mask_restore disabled
will give the better blending of "Implement masking during sampling".

* revert changes of a7be43ba in change_image_editor_mode

* fix ui_functions.change_image_editor_mode by adding gr.update to the end of the list it returns

* revert inserted newlines and whitespaces to match format of previous code

* improve caption of new option mask_restore

"Only modify regenerated parts of image"

* fix ui_functions.change_image_editor_mode by adding gr.update to the end of the list it returns

an old copy of the function exists in webui.py, this superflous function mistakenly was changed by the earlier commit b6a9e16b

* remove unused functions that are near duplicates of functions in ui_functions.py
2022-09-10 00:07:14 +03:00
ZeroCool940711
61b3ca7014 Merge remote-tracking branch 'origin/dev' into dev 2022-09-09 03:44:25 -07:00
ZeroCool940711
22b84afc87 Improvements to the txt2vid tab. 2022-09-09 03:30:43 -07:00
Tony Beeman
7d987cc922 Add info and stats to img2img 2022-09-09 01:33:28 -07:00
Tony Beeman
28d8827331 Display Info and Stats when render is complete, similar to what Gradio shows. 2022-09-09 01:33:28 -07:00
Tony Beeman
771a68e3e5 Fixed the path in webui_streamlit.py 2022-09-09 01:03:34 -07:00
Tony Beeman
ca8ee265cd Made sure img2txt and img2img checkboxes respect YAML defaults 2022-09-08 23:50:33 -07:00
Tony Beeman
5ea27f3d0c Streamlit: Allow user defaults to be specified in a userconfig_streamlit.yaml file. 2022-09-08 22:58:25 -07:00
ZeroCool940711
e6027fe306 Added num_inference_steps to config file and fixed incorrectly calls to the config file from the txt2vid tab calling txt2img instead. 2022-09-08 05:51:22 -07:00
ZeroCool940711
ed84e1f1d8 - Added option to select custom models instead of just using the default one, if you want to use a custom model just place your .ckpt file in "models/custom" and the UI will detect it and let you switch between stable diffusion and your custom model, make sure to give the filename a proper name that is easy to distinguish from other models because that name will be used on the UI.
- Implemented basic Text To Video tab, will continue to improve it as it is really basic right now.
- Improved the model loading, you now should see less frequently errors about it not been loaded correctly.
2022-09-08 05:33:45 -07:00
Hafiidz
0e30d4fdd9 separate css to external file 2022-09-08 10:10:03 +08:00
ZeroCool940711
c0e02742e3 Added try and except block to handle basic errors like StopException which is raised by streamlit when you hit the stop button and KeyError which happens also when stopping the generation because it tries to check the model at the end which is not loaded at that time, this can be ignored and so thats the reason for the exception. 2022-09-07 18:03:39 -07:00
ZeroCool940711
1d4351ce91 Added option to set default sampler name from config file, will be useful for those wanting to change the default sampler and have it persist even when closing the UI and opening it again. 2022-09-07 16:58:19 -07:00
Thomas Mello
a81671d673
fix: stack overflow during recursion call (#784) 2022-09-07 22:14:46 +03:00
Thomas Mello
388a765162 fix: regex escape characters 2022-09-07 18:05:16 +03:00
Hafiidz
432ebdad29 added auto conversion for 4 channel PNG to RGB 2022-09-07 14:29:49 +08:00
Hafiidz
03a3c149ba added placeholder text similar to gradio 2022-09-07 13:32:23 +08:00
ZeroCool940711
5b79864d43 Moved streamlit files to their proper location, before they were incorrectly added to the repository root folder. 2022-09-06 20:20:29 -07:00