Upload mmalaya_arch.py

mmalaya_arch.py

@@ -300,7 +300,11 @@ class MMAlayaMetaForCausalLM(ABC):
             input_ids.extend(x[offset:])
 
         # 加载generate stop条件
-        stopping_criteria = KeywordsStoppingCriteria([conv.sep2], tokenizer, input_ids)
+        stopping_criteria = KeywordsStoppingCriteria(
+            [conv.sep2], 
+            tokenizer, 
+            torch.tensor(input_ids, dtype=torch.long),
+            )
         # 加载图像
         image_processor = model.get_vision_tower().image_processor
         image_tensor = image_processor(image, return_tensors='pt')['pixel_values'].half().cuda()