XY PLot Refiner ClipSkip Printouts Fix

This commit is contained in:
TSC
2023-09-05 19:13:21 -05:00
committed by GitHub
parent 5947885ee3
commit 545ccc012a

View File

@@ -1590,10 +1590,11 @@ class TSC_KSampler:
vae_name = "" vae_name = ""
return vae_name return vae_name
def get_clip_skip(X_type, Y_type, X_value, Y_value, cskip): def get_clip_skip(X_type, Y_type, X_value, Y_value, cskip, mode):
if "Clip Skip" in X_type: clip_type = "Clip Skip" if mode == "ckpt" else "Clip Skip (Refiner)"
if X_type == clip_type:
cskip = ", ".join(map(str, X_value)) cskip = ", ".join(map(str, X_value))
elif "Clip Skip" in Y_type: elif Y_type == clip_type:
cskip = ", ".join(map(str, Y_value)) cskip = ", ".join(map(str, Y_value))
elif cskip[1] != None: elif cskip[1] != None:
cskip = cskip[1] cskip = cskip[1]
@@ -1620,10 +1621,10 @@ class TSC_KSampler:
othr_type, othr_value = X_type, X_value.copy() othr_type, othr_value = X_type, X_value.copy()
else: else:
# Process as per original function if mode is "ckpt" # Process as per original function if mode is "ckpt"
clip_skip = get_clip_skip(X_type, Y_type, X_value, Y_value, clip_skip, mode)
if mode == "ckpt": if mode == "ckpt":
if vae_name: if vae_name:
vae_name = get_vae_name(X_type, Y_type, X_value, Y_value, vae_name) vae_name = get_vae_name(X_type, Y_type, X_value, Y_value, vae_name)
clip_skip = get_clip_skip(X_type, Y_type, X_value, Y_value, clip_skip)
return ckpt_name, clip_skip, vae_name return ckpt_name, clip_skip, vae_name
else: else:
# For refn mode # For refn mode