StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
1db7d21283
WebUI/modules
History
fumitaka.yano 1db7d21283 Subject:. 
Improvements to handle VAE filenames in generated image filenames

Body:.
1) Added new line 24 to import sd_vae module.
2) Added new method get_vae_filename at lines 340-349 to obtain the VAE filename to be used for image generation and further process it to extract only the filename by splitting it with a dot symbol.
3) Added a new lambda function 'vae_filename' at line 373 to handle VAE filenames.

Reason:.
A function was needed to get the VAE filename and handle it in the program.

Test:.
We tested whether we could use this new functionality to get the expected file names.
The correct behaviour was confirmed for the following commonly distributed VAE files.
vae-ft-mse-840000-ema-pruned.safetensors -> vae-ft-mse-840000-ema-pruned
anything-v4.0.vae.pt -> anything-v4.0

ruff response:.
There were no problems with the code I added.

There was a minor configuration error in a line I did not modify, but I did not modify it as it was not relevant to this modification.
Logged.
images.py:426:56: F841 [*] Local variable `_` is assigned to but never used
images.py:432:43: F841 [*] Local variable `_` is assigned to but never used

Impact:.
This change makes it easier to retrieve the VAE filename used for image generation and use it in the programme.
2023-05-23 15:56:08 +09:00
..
api Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
codeformer fix F541 f-string without any placeholders 2022-12-24 21:35:29 +02:00
hypernetworks sort hypernetworks and checkpoints by name 2023-03-28 20:03:57 +03:00
models/diffusion Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
textual_inversion Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
call_queue.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
cmd_args.py Merge branch 'experimental' into subpath-support 2023-05-01 11:59:21 -07:00
codeformer_model.py Filter out temporary files that will be generated if the download fails. 2023-02-24 00:42:29 +08:00
config_states.py Various UI fixes in config state tab 2023-03-29 18:35:25 -05:00
deepbooru_model.py Add option for float32 sampling with float16 UNet 2023-01-25 01:13:02 -05:00
deepbooru.py added an option to filter out deepbooru tags 2022-12-24 16:22:47 +03:00
devices.py rename CPU RNG to RNG source in settings, add infotext and parameters copypaste support to RNG source 2023-04-29 11:29:37 +03:00
errors.py better support for xformers flash attention on older versions of torch 2023-01-23 16:40:20 +03:00
esrgan_model_arch.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
esrgan_model.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
extensions.py Better checking of extension state from Git info 2023-03-29 18:32:29 -05:00
extra_networks_hypernet.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
extra_networks.py extra networks UI 2023-01-21 08:36:07 +03:00
extras.py Namespace metadata fields 2023-04-02 21:41:23 -05:00
face_restoration.py codeformer support 2022-09-07 12:32:28 +03:00
generation_parameters_copypaste.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
gfpgan_model.py add data-dir flag and set all user data directories based on it 2023-01-27 14:44:30 +01:00
hashes.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
images.py Subject:. 2023-05-23 15:56:08 +09:00
img2img.py print PIL.UnidentifiedImageError 2023-05-03 14:28:59 +09:00
import_hook.py add a comment for disable xformers hack 2022-12-24 07:57:56 +03:00
interrogate.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
localization.py Merge pull request #10335 from akx/l10n-dis-take-2 2023-05-13 20:46:50 +03:00
lowvram.py Add support for the Variations models (unclip-h and unclip-l) 2023-03-25 21:03:07 -04:00
mac_specific.py Fix generation with k-diffusion/UniPC on x64 Macs 2023-05-09 01:10:13 -04:00
masking.py Fix non-square full resolution inpainting. 2022-10-30 15:33:02 -07:00
memmon.py attempt to fix memory monitor with multiple CUDA devices 2023-03-12 21:04:17 +03:00
modelloader.py fix upscalers disappearing after the user reloads UI 2023-05-13 20:21:11 +03:00
ngrok.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
paths_internal.py Save/restore working webui/extension configs 2023-03-29 16:55:33 -05:00
paths.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
postprocessing.py Merge branch 'dev' into master 2023-04-29 17:02:14 +03:00
processing.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
progress.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
prompt_parser.py add BREAK keyword to end current text chunk and start the next 2023-01-15 22:29:53 +03:00
realesrgan_model.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
safe.py Merge pull request #10339 from catboxanon/bf16 2023-05-13 20:21:39 +03:00
script_callbacks.py remove unwanted changes from #8789 2023-04-29 19:05:43 +03:00
script_loading.py use python importlib to load and execute extension modules 2023-01-27 14:44:34 +01:00
scripts_auto_postprocessing.py add an option to enable sections from extras tab in txt2img/img2img 2023-01-26 23:31:32 +03:00
scripts_postprocessing.py initial gradio 3.22 support 2023-03-25 09:00:36 +03:00
scripts.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
sd_disable_initialization.py make it possible to load SD1 checkpoints without CLIP 2023-02-05 11:21:00 +03:00
sd_hijack_checkpoint.py add option to show/hide warnings 2023-01-18 23:04:24 +03:00
sd_hijack_clip_old.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
sd_hijack_clip.py add BREAK keyword to end current text chunk and start the next 2023-01-15 22:29:53 +03:00
sd_hijack_inpainting.py add missing import 2023-02-08 07:10:13 -05:00
sd_hijack_ip2p.py fix broken calls to find_checkpoint_config 2023-01-27 22:43:08 +03:00
sd_hijack_open_clip.py Add support Stable Diffusion 2.0 2022-11-26 16:10:46 +03:00
sd_hijack_optimizations.py Fix for Unet NaNs 2023-05-08 08:16:56 -04:00
sd_hijack_unet.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
sd_hijack_utils.py Fix full previews, --no-half-vae 2023-01-26 01:43:35 -05:00
sd_hijack_xlmr.py alt-diffusion integration 2022-12-31 18:06:35 +03:00
sd_hijack.py sdp_attnblock_forward hijack 2023-03-10 22:48:41 +05:00
sd_models_config.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
sd_models.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
sd_samplers_common.py rename CPU RNG to RNG source in settings, add infotext and parameters copypaste support to RNG source 2023-04-29 11:29:37 +03:00
sd_samplers_compvis.py Add support for the Variations models (unclip-h and unclip-l) 2023-03-25 21:03:07 -04:00
sd_samplers_kdiffusion.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
sd_samplers.py UniPC does not support img2img (for now) 2023-02-10 05:00:54 -08:00
sd_vae_approx.py modules/sd_vae_approx.py: fix VAE-approx path 2023-03-11 20:00:12 +08:00
sd_vae.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
shared_items.py remove unneeded return from #7583 2023-02-19 12:23:40 +03:00
shared.py Merge pull request #10324 from catboxanon/offline 2023-05-13 20:22:09 +03:00
styles.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
sub_quadratic_attention.py Add UI setting for upcasting attention to float32 2023-01-25 01:13:04 -05:00
timer.py display correct timings after restarting UI 2023-03-12 21:25:22 +03:00
txt2img.py Revert "Fix gallery not being refreshed correctly" 2023-05-04 14:08:22 +08:00
ui_common.py bump gradio to 3.27 2023-04-29 09:17:35 +03:00
ui_components.py Allow user input for gradio theme selection 2023-04-29 12:52:09 +03:00
ui_extensions.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
ui_extra_networks_checkpoints.py save previews for extra networks in the selected format 2023-03-11 15:46:20 +03:00
ui_extra_networks_hypernets.py save previews for extra networks in the selected format 2023-03-11 15:46:20 +03:00
ui_extra_networks_textual_inversion.py save previews for extra networks in the selected format 2023-03-11 15:46:20 +03:00
ui_extra_networks.py Fix up string formatting/concatenation to f-strings where feasible 2023-05-09 22:25:39 +03:00
ui_postprocessing.py Merge branch 'dev' into ui-config-tabs 2023-04-29 17:42:57 +03:00
ui_tempdir.py refresh fix 2023-05-09 19:06:00 +08:00
ui.py Merge pull request #10335 from akx/l10n-dis-take-2 2023-05-13 20:46:50 +03:00
upscaler.py add data-dir flag and set all user data directories based on it 2023-01-27 14:44:30 +01:00
xlmr.py fix bugs 2022-11-30 14:56:12 +08:00