Spaces:

shi-labs
/

VCoder

Build error

praeclarumjj3 commited on Dec 21, 2023

Commit

2a2a86d

•

1 Parent(s): 9c4915e

Update vcoder_llava/model/vcoder_ds_llava_arch.py

Files changed (1) hide show

vcoder_llava/model/vcoder_ds_llava_arch.py CHANGED Viewed

@@ -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]

         cur_seg_idx = 0
         cur_depth_idx = 0
         for batch_idx, cur_input_ids in enumerate(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]