lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-21 06:48:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

remove old stuff

Browse Source
This commit is contained in:
layerdiffusion
2024-08-05 12:01:11 -07:00
parent 24cfce26dc
commit a9f4957a27
3 changed files with 41 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
modules/sd_models.py
View File

@@ -379,29 +379,7 @@ def check_fp8(model):
def set_model_type(model, state_dict):
model.is_sd1 = False
model.is_sd2 = False
model.is_sdxl = False
model.is_ssd = False
model.is_sd3 = False
if "model.diffusion_model.x_embedder.proj.weight" in state_dict:
model.is_sd3 = True
model.model_type = ModelType.SD3
elif hasattr(model, 'conditioner'):
model.is_sdxl = True
if 'model.diffusion_model.middle_block.1.transformer_blocks.0.attn1.to_q.weight' not in state_dict.keys():
model.is_ssd = True
model.model_type = ModelType.SSD
else:
model.model_type = ModelType.SDXL
elif hasattr(model.cond_stage_model, 'model'):
model.is_sd2 = True
model.model_type = ModelType.SD2
else:
model.is_sd1 = True
model.model_type = ModelType.SD1
pass
def set_model_fields(model):