Update vcoder_llava/model/vcoder_ds_llava_arch.py

vcoder_llava/model/vcoder_ds_llava_arch.py

@@ -178,7 +178,6 @@ class VCoderDSLlavaMetaForCausalLM(ABC):
         cur_seg_idx = 0
         cur_depth_idx = 0
         for batch_idx, cur_input_ids in enumerate(input_ids):
-            print(cur_input_ids)
             if (cur_input_ids == IMAGE_TOKEN_INDEX).sum() == 0 and (cur_input_ids == SEG_TOKEN_INDEX).sum() == 0:
                 # FIXME: this is a hacky fix, for deepspeed zero3 to work
                 cur_image_features = image_features[cur_image_idx]