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 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|