Upload visual_encoder.py with huggingface_hub
Browse files- 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 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|