Thatphum commited on
Commit
e2f1075
·
verified ·
1 Parent(s): 689220a

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. modeling_GOT.py +2 -2
modeling_GOT.py CHANGED
@@ -271,8 +271,8 @@ class GOTQwenModel(Qwen2Model):
271
 
272
  if use_im_start_end:
273
  if (cur_input_ids == im_start_token).sum() != (cur_input_ids == im_end_token).sum():
274
- raise ValueError("The number of image start tokens and image end tokens should be the same.")
275
-
276
  image_start_tokens = torch.where(cur_input_ids == im_start_token)[0]
277
  for image_start_token_pos, per_cur_image_features in zip(image_start_tokens, cur_image_features):
278
  per_cur_image_features = per_cur_image_features.to(device=cur_input_embeds.device)
 
271
 
272
  if use_im_start_end:
273
  if (cur_input_ids == im_start_token).sum() != (cur_input_ids == im_end_token).sum():
274
+ print("Warning Please start, end image token not align")
275
+
276
  image_start_tokens = torch.where(cur_input_ids == im_start_token)[0]
277
  for image_start_token_pos, per_cur_image_features in zip(image_start_tokens, cur_image_features):
278
  per_cur_image_features = per_cur_image_features.to(device=cur_input_embeds.device)