tamoghna
/

encoder-decoder-eng2bn-eng2hi

small_transformer

Model card Files Files and versions

tamoghna commited on Nov 7, 2025

Commit

496a8a8

·

verified ·

1 Parent(s): 31831aa

Update modeling.py

Files changed (1) hide show

modeling.py +1 -0

modeling.py CHANGED Viewed

@@ -129,6 +129,7 @@ import torch.nn as nn
 from typing import Optional, Tuple
 from transformers import PreTrainedModel
 from transformers.modeling_outputs import Seq2SeqLMOutput
 class SmallTransformerConfig(PretrainedConfig):
     model_type = "small_transformer"

 from typing import Optional, Tuple
 from transformers import PreTrainedModel
 from transformers.modeling_outputs import Seq2SeqLMOutput
+from transformers.configuration_utils import PretrainedConfig
 class SmallTransformerConfig(PretrainedConfig):
     model_type = "small_transformer"