hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-04 03:10:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

support for SD3: infinite prompt length, token counting

Browse Source
This commit is contained in:
AUTOMATIC1111
2024-06-26 23:22:00 +03:00
parent a8fba9af35
commit d686e73daa
6 changed files with 278 additions and 139 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
modules/sd_models.py
View File

@@ -722,7 +722,12 @@ def get_empty_cond(sd_model):
d = sd_model.get_learned_conditioning([""])
return d['crossattn']
else:
return sd_model.cond_stage_model([""])
d = sd_model.cond_stage_model([""])
if isinstance(d, dict):
d = d['crossattn']
return d
def send_model_to_cpu(m):