Update transformer_mt_roberta/modeling_transformer_final.py

transformer_mt_roberta/modeling_transformer_final.py

@@ -91,7 +91,7 @@ class TransfomerEncoderDecoderModel(nn.Module):
 
         self.dropout = nn.Dropout(self.dropout_rate)
 
-        self.encoder = AutoModelForMaskedLM.from_pretrained("flax-community/roberta_base_danish", output_hidden_states=True)
+        self.encoder = AutoModelForMaskedLM.from_pretrained("flax-community/roberta-base-danish", output_hidden_states=True)
 
         self.decoder_layers = nn.ModuleList([TransformerDecoderLayer(hidden = self.hidden,
                                                                     num_heads = self.num_heads,