aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_models_config.py
diff options
context:
space:
mode:
authorsuperhero-7 <537093830@qq.com>2023-10-01 04:25:19 +0000
committersuperhero-7 <537093830@qq.com>2023-10-01 04:25:19 +0000
commit2d947175b902d6838c803036d9757e7d3226b41d (patch)
treed62d9f599c94734d5e40b5ed711be13e0d4e081d /modules/sd_models_config.py
parentf8f4ff2bb8f56877dede466934dd8ddf25c21063 (diff)
downloadstable-diffusion-webui-gfx803-2d947175b902d6838c803036d9757e7d3226b41d.tar.gz
stable-diffusion-webui-gfx803-2d947175b902d6838c803036d9757e7d3226b41d.tar.bz2
stable-diffusion-webui-gfx803-2d947175b902d6838c803036d9757e7d3226b41d.zip
fix linter issues
Diffstat (limited to 'modules/sd_models_config.py')
-rw-r--r--modules/sd_models_config.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/modules/sd_models_config.py b/modules/sd_models_config.py
index 9ba89dfc..deab2f6e 100644
--- a/modules/sd_models_config.py
+++ b/modules/sd_models_config.py
@@ -95,8 +95,7 @@ def guess_model_config_from_state_dict(sd, filename):
if diffusion_model_input.shape[1] == 8:
return config_instruct_pix2pix
-
- # import pdb; pdb.set_trace()
+
if sd.get('cond_stage_model.roberta.embeddings.word_embeddings.weight', None) is not None:
if sd.get('cond_stage_model.transformation.weight').size()[0] == 1024:
return config_alt_diffusion_m18