chendl commited on
Commit
013e410
·
1 Parent(s): ca42c83

update cap

Browse files
multimodal/open_flamingo/chat/conversation.py CHANGED
@@ -458,7 +458,7 @@ class Chat:
458
  # f"### Assistant: {tokenizer.decode(outputs[0, input_ids.shape[1]:], skip_special_tokens=True).strip()}")
459
  output_text = self.tokenizer.decode(text_outputs[0])
460
  print(output_text)
461
- output_text = re.findall(r'Assistant:(.+)', output_text)
462
  print(output_text)
463
 
464
  return output_text, out_image
 
458
  # f"### Assistant: {tokenizer.decode(outputs[0, input_ids.shape[1]:], skip_special_tokens=True).strip()}")
459
  output_text = self.tokenizer.decode(text_outputs[0])
460
  print(output_text)
461
+ output_text = re.findall(r'Assistant:(.+)', output_text)[-1]
462
  print(output_text)
463
 
464
  return output_text, out_image