diff options
Diffstat (limited to 'models/clip/tokenizer.py')
-rw-r--r-- | models/clip/tokenizer.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/models/clip/tokenizer.py b/models/clip/tokenizer.py index 11a3df0..4e97ab5 100644 --- a/models/clip/tokenizer.py +++ b/models/clip/tokenizer.py | |||
@@ -48,7 +48,7 @@ def shuffle_none(tokens: list[int]): | |||
48 | 48 | ||
49 | 49 | ||
50 | def shuffle_auto(tokens: list[int]): | 50 | def shuffle_auto(tokens: list[int]): |
51 | if len(tokens) >= 4: | 51 | if len(tokens) >= 5: |
52 | return shuffle_between(tokens) | 52 | return shuffle_between(tokens) |
53 | if len(tokens) >= 3: | 53 | if len(tokens) >= 3: |
54 | return shuffle_trailing(tokens) | 54 | return shuffle_trailing(tokens) |