Minthy
/

ToriiGate-0.5

Image-Text-to-Text

Model card Files Files and versions

Minthy commited on 20 days ago

Commit

667e771

·

verified ·

1 Parent(s): 78f6e30

Update scripts/caption_distributed.py

Files changed (1) hide show

scripts/caption_distributed.py +1 -1

scripts/caption_distributed.py CHANGED Viewed

@@ -141,7 +141,7 @@ def prepare_messages(item: Dict[str, Any], image_data: str) -> list:
         add_tags=ADD_TAGS,
         add_characters=ADD_CHAR_LIST,
         add_char_tags=ADD_CHARS_TAGS,
-        add_descritpion=ADD_CHARS_DESCR,
         underscores_replace=False
     )

         add_tags=ADD_TAGS,
         add_characters=ADD_CHAR_LIST,
         add_char_tags=ADD_CHARS_TAGS,
+        add_description=ADD_CHARS_DESCR,
         underscores_replace=False
     )