teowu commited on
Commit
946fc84
1 Parent(s): 24d1a9a

Upload visual_encoder.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. visual_encoder.py +1 -9
visual_encoder.py CHANGED
@@ -917,12 +917,4 @@ class MplugOwlVisualAbstractorModel(PreTrainedModel):
917
  hidden_states=encoder_outputs.hidden_states,
918
  )
919
 
920
-
921
- if __name__ == "__main__":
922
- from configuration_mplug_owl2 import MPLUGOwl2Config
923
- config = MPLUGOwl2Config()
924
- visual_model = MplugOwlVisionModel(config.visual_config["visual_model"])
925
- print(visual_model)
926
-
927
- abstractor_module = MplugOwlVisualAbstractorModel(config.visual_config["visual_abstractor"], config.hidden_size)
928
- print(abstractor_module)
 
917
  hidden_states=encoder_outputs.hidden_states,
918
  )
919
 
920
+