lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-13 17:09:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

add some misc fix related to info text

Browse Source
This commit is contained in:
layerdiffusion
2024-08-27 07:51:28 -07:00
parent d1121baf80
commit 9c03493262
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
modules/infotext_utils.py
View File

@@ -277,6 +277,26 @@ Steps: 20, Sampler: Euler a, CFG scale: 7, Seed: 965400086, Size: 512x512, Model
else:
prompt += ("" if prompt == "" else "\n") + line
if 'Civitai' in lastline and 'FLUX' in lastline:
# Civitai really like to add random Clip skip to Flux metadata, where Clip skip is not a thing.
lastline = lastline.replace('Clip skip: 0, ', '')
lastline = lastline.replace('Clip skip: 1, ', '')
lastline = lastline.replace('Clip skip: 2, ', '')
lastline = lastline.replace('Clip skip: 3, ', '')
lastline = lastline.replace('Clip skip: 4, ', '')
lastline = lastline.replace('Clip skip: 5, ', '')
lastline = lastline.replace('Clip skip: 6, ', '')
lastline = lastline.replace('Clip skip: 7, ', '')
lastline = lastline.replace('Clip skip: 8, ', '')
# Civitai also add Sampler: Undefined
lastline = lastline.replace('Sampler: Undefined, ', 'Sampler: Euler, Schedule type: Simple, ') # <- by lllyasviel, seem to give similar results to Civitai "Undefined" Sampler
# Civitai also confuse CFG scale and Distilled CFG Scale
lastline = lastline.replace('CFG scale: ', 'CFG scale: 1, Distilled CFG Scale: ')
print('Applied Forge Fix to broken Civitai Flux Meta.')
for k, v in re_param.findall(lastline):
try:
if v[0] == '"' and v[-1] == '"':