k4d3 commited on
Commit
f8b95db
1 Parent(s): 51b7ba7

Signed-off-by: Balazs Horvath <acsipont@gmail.com>

Files changed (1) hide show
  1. joy +1 -1
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