Python BartForCausalLM Examples

Example #1

File: test_modeling_encoder_decoder.py Project: ankitshah009/pytorch-pretrained-BERT

 def get_encoder_decoder_model(self, config, decoder_config):
     encoder_model = BertModel(config)
     decoder_model = BartForCausalLM(decoder_config)
     return encoder_model, decoder_model

Example #2

File: test_modeling_vision_encoder_decoder.py Project: calpt/transformers

 def get_encoder_decoder_model(self, config, decoder_config):
     encoder_model = SwinModel(config).eval()
     decoder_model = BartForCausalLM(decoder_config).eval()
     return encoder_model, decoder_model