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) |