Update modeling_motif.py

modeling_motif.py

@@ -1315,10 +1315,6 @@ class MotifModel(MotifPreTrainedModel):
                 )
 
             hidden_states = layer_outputs[0]
-            
-
-            if self.use_pipeline and idx in self.split_layers:
-                hidden_states = torch.moreh.pipeline_assign(hidden_states)
 
             if use_cache:
                 next_decoder_cache = layer_outputs[2 if output_attentions else 1]