Upload visual_encoder.py with huggingface_hub
Browse files- visual_encoder.py +1 -0
visual_encoder.py
CHANGED
@@ -690,6 +690,7 @@ class MplugOwlVisualAbstractorLayer(nn.Module):
|
|
690 |
|
691 |
|
692 |
class MplugOwlVisualAbstractorEncoder(nn.Module):
|
|
|
693 |
def __init__(self, config):
|
694 |
super().__init__()
|
695 |
self.config = config
|
|
|
690 |
|
691 |
|
692 |
class MplugOwlVisualAbstractorEncoder(nn.Module):
|
693 |
+
_no_split_modules = ["MplugOwlVisualAbstractorLayer"]
|
694 |
def __init__(self, config):
|
695 |
super().__init__()
|
696 |
self.config = config
|