andrewqian123 commited on
Commit
61657ef
·
verified ·
1 Parent(s): 2033829

Update modeling_minicpmv.py

Browse files
Files changed (1) hide show
  1. modeling_minicpmv.py +2 -2
modeling_minicpmv.py CHANGED
@@ -304,7 +304,7 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
304
  copy_msgs = deepcopy(msgs[ind])
305
  copy_msgs[0]["content"] = [image, copy_msgs[0]["content"]]
306
 
307
- imagelist = []
308
  for i, msg in enumerate(copy_msgs):
309
  role = msg["role"]
310
  content = msg["content"]
@@ -316,7 +316,7 @@ class MiniCPMV(MiniCPMVPreTrainedModel):
316
  cur_msgs = []
317
  for c in content:
318
  if isinstance(c, Image.Image):
319
- imagelist.append(c)
320
  cur_msgs.append("(<image>./</image>)")
321
  elif isinstance(c, str):
322
  cur_msgs.append(c)
 
304
  copy_msgs = deepcopy(msgs[ind])
305
  copy_msgs[0]["content"] = [image, copy_msgs[0]["content"]]
306
 
307
+ imagelist = image
308
  for i, msg in enumerate(copy_msgs):
309
  role = msg["role"]
310
  content = msg["content"]
 
316
  cur_msgs = []
317
  for c in content:
318
  if isinstance(c, Image.Image):
319
+ imagelist = c
320
  cur_msgs.append("(<image>./</image>)")
321
  elif isinstance(c, str):
322
  cur_msgs.append(c)