Ejemplos de BartForCausalLM en Python

Ejemplo n.º 1

Archivo: test_modeling_encoder_decoder.py Proyecto: 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

Ejemplo n.º 2

Archivo: test_modeling_vision_encoder_decoder.py Proyecto: 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