Ignore negative keep_tags values

2023-09-20 19:50:34 -03:00 · 2023-09-20 19:50:34 -03:00 · 99a0431d0f
parent 43984f2ad3
commit 99a0431d0f
1 changed files with 1 additions and 0 deletions
--- a/data/image_train_item.py
+++ b/data/image_train_item.py
@ -113,6 +113,7 @@ class ImageCaption:
    @staticmethod
    def __get_shuffled_tags(seed: int, tags: list[str], keep_tags: int) -> str:
        tags = tags.copy()
+        keep_tags = min(keep_tags, 0)

        if len(tags) > keep_tags:
            fixed_tags = tags[:keep_tags]