From 3aa7bd746a9f70733a451c8d3bccddffb797dd19 Mon Sep 17 00:00:00 2001 From: kaidu Date: Fri, 22 Nov 2024 23:51:57 +0100 Subject: [PATCH] add high strengh mode --- nodes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nodes.py b/nodes.py index f4a9ce6..4a9fb6e 100644 --- a/nodes.py +++ b/nodes.py @@ -32,8 +32,8 @@ def automerge(tensor, threshold): newTensor.append(torch.stack(tokens)) return torch.stack(newTensor) -STRENGTHS = ["very high", "medium", "low", "very low"] -STRENGTHS_VALUES = [1,3,4,5] +STRENGTHS = ["very high", "high", "medium", "low", "very low"] +STRENGTHS_VALUES = [1,2, 3,4,5] class StyleModelApplySimple: @classmethod