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
|