ok..
Browse filesSigned-off-by: Balazs Horvath <acsipont@gmail.com>
joy
CHANGED
@@ -362,7 +362,7 @@ class JoyCaptionModel:
|
|
362 |
all(count == 1 for count in word_counts.values()) and
|
363 |
sentence_count > 1):
|
364 |
return caption
|
365 |
-
print("Generated caption is invalid. Retrying
|
366 |
|
367 |
def _get_prompt_string(self, caption_type, caption_tone, caption_length):
|
368 |
length = None if caption_length == "any" else caption_length
|
|
|
362 |
all(count == 1 for count in word_counts.values()) and
|
363 |
sentence_count > 1):
|
364 |
return caption
|
365 |
+
print(f"Generated caption is invalid. Retrying...\nCaption: {caption!r}")
|
366 |
|
367 |
def _get_prompt_string(self, caption_type, caption_tone, caption_length):
|
368 |
length = None if caption_length == "any" else caption_length
|