teowu commited on
Commit
4384eea
1 Parent(s): ba24430

Upload visual_encoder.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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