See megatron.model.transformer.set_input_tensor()
(self, input_tensor)
| 42 | args.num_layers)) |
| 43 | |
| 44 | def set_input_tensor(self, input_tensor): |
| 45 | """See megatron.model.transformer.set_input_tensor()""" |
| 46 | self.language_model.set_input_tensor(input_tensor) |
| 47 | |
| 48 | def forward( |
| 49 | self, |