StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
803d44c474
WebUI/modules
History
butaixianran 803d44c474
Fix None type error for TI module 
When user using model_name.png as a preview image, textural_inversion.py still treat it as an embeding, and didn't handle its error, just let python throw out an None type error like following:
```bash
  File "D:\Work\Dev\AI\stable-diffusion-webui\modules\textual_inversion\textual_inversion.py", line 155, in load_from_file
    name = data.get('name', name)
AttributeError: 'NoneType' object has no attribute 'get'
```

With just a simple `if data:` checking as following, there will be no error, breaks nothing, and now this module can works fine with user's preview images.
Old code:  
```python
                data = extract_image_data_embed(embed_image)
                name = data.get('name', name)
```
New code:  
```python
                data = extract_image_data_embed(embed_image)
                if data:
                    name = data.get('name', name)
                else:
                    # if data is None, means this is not an embeding, just a preview image
                    return
```

Also, since there is no more errors on textual inversion module, from now on, extra network can set "model_name.png" as preview image for embedings.
2023-03-25 02:05:00 +08:00
..
api small diff whitespace cleanup 2023-03-11 14:34:56 -05:00
codeformer fix F541 f-string without any placeholders 2022-12-24 21:35:29 +02:00
hypernetworks Merge branch 'master' into weighted-learning 2023-02-19 12:41:29 +03:00
models/diffusion add pbar to unipc 2023-03-13 12:35:30 -04:00
textual_inversion Fix None type error for TI module 2023-03-25 02:05:00 +08:00
call_queue.py big rework of progressbar/preview system to allow multiple users to prompts at the same time and do not get previews of each other 2023-01-15 18:51:04 +03:00
codeformer_model.py Filter out temporary files that will be generated if the download fails. 2023-02-24 00:42:29 +08: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 Refactor Mac specific code to a separate file 2023-02-01 14:05:56 -05: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 add missing import 2023-02-08 07:08:09 -05:00
esrgan_model.py Revert "MPS Upscalers Fix" 2022-11-17 00:08:21 -05:00
extensions.py git 3.1.30 api change 2023-02-25 19:15:06 +00:00
extra_networks_hypernet.py add dropdowns in settings for hypernets and loras 2023-01-28 17:18:47 +03:00
extra_networks.py extra networks UI 2023-01-21 08:36:07 +03:00
extras.py Merge pull request #7240 from Unstackd/master 2023-01-28 10:52:28 +03:00
face_restoration.py codeformer support 2022-09-07 12:32:28 +03:00
generation_parameters_copypaste.py Merge pull request #7818 from space-nuko/extension-paste-field-names 2023-03-11 16:44:20 +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 add --no-hashing 2023-02-04 11:38:56 +03:00
images.py update fullfn properly 2023-03-12 12:36:04 -07:00
img2img.py convert rgba to rgb 2023-02-08 07:03:36 -05:00
import_hook.py add a comment for disable xformers hack 2022-12-24 07:57:56 +03:00
interrogate.py add data-dir flag and set all user data directories based on it 2023-01-27 14:44:30 +01:00
localization.py add ability to create extensions that add localizations 2022-11-06 09:02:35 +03:00
lowvram.py unload depth model if medvram/lowvram enabled 2022-12-10 11:02:47 -05:00
mac_specific.py Remove test, use bool tensor fix by default 2023-03-11 17:35:17 -05: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 use lowercase name for sorting; keep UpscalerLanczos and UpscalerNearest at the start of the list with UpscalerNone 2023-03-06 13:21:42 -05:00
ngrok.py Correct singleton comparisons 2022-12-14 20:59:33 +02:00
paths.py don't replace regular --help with new paths.py parser help 2023-01-28 02:44:12 -05:00
postprocessing.py a possible fix for broken image upscaling 2023-01-23 17:59:58 +03:00
processing.py initialize extra_network_data before use 2023-03-13 17:37:29 +02:00
progress.py add progress bar to modelmerger 2023-01-19 09:25:37 +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 Refactor conditional casting, fix upscalers 2023-01-28 04:16:25 -05:00
safe.py Merge pull request #5992 from yuvalabou/F541 2022-12-25 09:16:08 +03:00
script_callbacks.py rename CFGDenoiserParams fields for #8064 2023-03-11 14:52:29 +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 add an option to enable sections from extras tab in txt2img/img2img 2023-01-26 23:31:32 +03:00
scripts.py Merge pull request #7818 from space-nuko/extension-paste-field-names 2023-03-11 16:44:20 +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 CLIP hijack rework 2023-01-07 01:46:13 +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 sdp_attnblock_forward hijack 2023-03-10 22:48:41 +05:00
sd_hijack_unet.py Apply hijacks in ddpm_edit for upcast sampling 2023-02-07 22:53:45 -05: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 automatically detect v-parameterization for SD2 checkpoints 2023-01-28 15:24:41 +03:00
sd_models.py fix an error loading Lora with empty values in metadata 2023-03-14 11:22:29 +03:00
sd_samplers_common.py Refactor Mac specific code to a separate file 2023-02-01 14:05:56 -05:00
sd_samplers_compvis.py add UniPC options to infotext 2023-03-11 12:09:36 +03:00
sd_samplers_kdiffusion.py rename CFGDenoiserParams fields for #8064 2023-03-11 14:52:29 +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 add data-dir flag and set all user data directories based on it 2023-01-27 14:44:30 +01:00
shared_items.py remove unneeded return from #7583 2023-02-19 12:23:40 +03:00
shared.py force refresh tqdm before close 2023-03-12 09:19:23 -04:00
styles.py change style dropdowns to multiselect 2023-01-14 14:56: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 make it so that setting options in pasted infotext (like Clip Skip and ENSD) do not get applied directly and instead are added as temporary overrides 2023-01-30 00:25:30 +03:00
ui_common.py Allow extensions to declare paste fields for "Send to X" buttons 2023-02-14 03:55:42 -08:00
ui_components.py add an option to enable sections from extras tab in txt2img/img2img 2023-01-26 23:31:32 +03:00
ui_extensions.py change extension index link to the new dedicated repo instead of wiki 2023-03-12 17:20:17 +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 Update ui_extra_networks.py 2023-03-14 15:36:08 +09:00
ui_postprocessing.py rework extras tab to use script system 2023-01-23 09:24:43 +03:00
ui_tempdir.py add support for running with gradio 3.9 installed 2023-01-03 14:18:48 +03:00
ui.py Merge pull request #8492 from zhanghua000/absolute-path 2023-03-12 08:55:15 +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