stanfordnlp
/

mrt5-small

text2text-generation

Model card Files Files and versions

juliekallini commited on Mar 23

Commit

09f3e2b

·

verified ·

1 Parent(s): 6e3412e

Update modeling_mrt5.py

Files changed (1) hide show

modeling_mrt5.py +2 -2

modeling_mrt5.py CHANGED Viewed

@@ -8,7 +8,7 @@ import torch
 import copy
 import numpy as np
 from torch import nn
-from models.modeling_t5 import (
     T5Attention,
     T5LayerNorm,
     T5LayerFF,
@@ -16,7 +16,7 @@ from models.modeling_t5 import (
     T5ForConditionalGeneration,
     softmax1,
 )
-from configuration_mrt5 import MrT5Config
 from transformers.modeling_outputs import (
     BaseModelOutput,
     BaseModelOutputWithPastAndCrossAttentions,

 import copy
 import numpy as np
 from torch import nn
+from .modeling_t5 import (
     T5Attention,
     T5LayerNorm,
     T5LayerFF,
     T5ForConditionalGeneration,
     softmax1,
 )
+from .configuration_mrt5 import MrT5Config
 from transformers.modeling_outputs import (
     BaseModelOutput,
     BaseModelOutputWithPastAndCrossAttentions,