chendl commited on
Commit
42f12c1
·
1 Parent(s): ceb509e

update cap

Browse files
multimodal/open_flamingo/chat/conversation.py CHANGED
@@ -319,8 +319,7 @@ class Chat:
319
  # else:
320
  # conv.append_message(conv.roles[0], text)
321
 
322
- def answer(self, conv, img_list, radio, text_input, model_name, max_new_tokens=200, num_beams=5, min_length=1,
323
- top_p=0.9,
324
  repetition_penalty=1.0, length_penalty=1, temperature=1, max_length=2000):
325
  # conv.append_message(conv.roles[1], None)
326
  # embs = self.get_context_emb(conv, img_list)
 
319
  # else:
320
  # conv.append_message(conv.roles[0], text)
321
 
322
+ def answer(self, conv, img_list, radio, text_input, model_name, max_new_tokens=200, num_beams=5, min_length=1, top_p=0.9,
 
323
  repetition_penalty=1.0, length_penalty=1, temperature=1, max_length=2000):
324
  # conv.append_message(conv.roles[1], None)
325
  # embs = self.get_context_emb(conv, img_list)