add high strengh mode

2024-11-22 23:51:57 +01:00
parent b84294d70d
commit 3aa7bd746a
1 changed files with 2 additions and 2 deletions
@@ -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