bingwork commited on
Commit
0db747a
1 Parent(s): fdbceb2

Upload mmalaya_arch.py

Browse files
Files changed (1) hide show
  1. mmalaya_arch.py +5 -1
mmalaya_arch.py CHANGED
@@ -300,7 +300,11 @@ class MMAlayaMetaForCausalLM(ABC):
300
  input_ids.extend(x[offset:])
301
 
302
  # 加载generate stop条件
303
- stopping_criteria = KeywordsStoppingCriteria([conv.sep2], tokenizer, input_ids)
 
 
 
 
304
  # 加载图像
305
  image_processor = model.get_vision_tower().image_processor
306
  image_tensor = image_processor(image, return_tensors='pt')['pixel_values'].half().cuda()
 
300
  input_ids.extend(x[offset:])
301
 
302
  # 加载generate stop条件
303
+ stopping_criteria = KeywordsStoppingCriteria(
304
+ [conv.sep2],
305
+ tokenizer,
306
+ torch.tensor(input_ids, dtype=torch.long),
307
+ )
308
  # 加载图像
309
  image_processor = model.get_vision_tower().image_processor
310
  image_tensor = image_processor(image, return_tensors='pt')['pixel_values'].half().cuda()