Removed dependency on icecream

modeling_mplug_owl2.py

@@ -37,7 +37,6 @@ from .modeling_attn_mask_utils import _prepare_4d_causal_attention_mask
 IGNORE_INDEX = -100
 IMAGE_TOKEN_INDEX = -200
 DEFAULT_IMAGE_TOKEN = "<|image|>"
-from icecream import ic
 
 def tokenizer_image_token(prompt, tokenizer, image_token_index=IMAGE_TOKEN_INDEX, return_tensors=None):
     prompt_chunks = [tokenizer(chunk).input_ids if len(chunk) > 0 else [] for chunk in prompt.split(DEFAULT_IMAGE_TOKEN)]
@@ -391,7 +390,6 @@ replace_llama_modality_adaptive()
 
 if __name__ == "__main__":
     config = MPLUGOwl2Config.from_pretrained('q-future/one-align')
-    from icecream import ic
     # config = MPLUGOwl2Config()
     model =  AutoModelForCausalLM(config)
     
@@ -403,10 +401,7 @@ if __name__ == "__main__":
     labels[labels < 0] = -100
     
     # image_feature = model.encode_images(images)
-    # ic(image_feature.shape)
     
     output = model(images=images, input_ids=input_ids, labels=labels)
-    ic(output.loss)
-    ic(output.logits.shape)
     
-    model.save_pretrained('/cpfs01/shared/public/test/tmp_owl')
+    model.save_pretrained('/cpfs01/shared/public/test/tmp_owl')

visual_encoder.py

@@ -9,7 +9,6 @@ import numpy as np
 import torch
 import torch.nn as nn
 import torch.utils.checkpoint
-from icecream import ic
 
 def get_abs_pos(abs_pos, tgt_size):
     # abs_pos: L, C
@@ -919,4 +918,4 @@ class MplugOwlVisualAbstractorModel(PreTrainedModel):
             hidden_states=encoder_outputs.hidden_states,
         )
     
-    
+