ai4bharat
/

indictrans2-indic-en-1B

text2text-generation

Model card Files Files and versions Community

jaygala24 commited on Sep 12, 2023

Commit

3cecfb6

•

1 Parent(s): 88a9a34

Update modeling_indictrans.py

Files changed (1) hide show

modeling_indictrans.py +2 -2

modeling_indictrans.py CHANGED Viewed

@@ -23,7 +23,7 @@ import torch.nn as nn
 from torch.nn import functional as F
 from transformers.activations import ACT2FN
-from transformers.deepspeed import is_deepspeed_zero3_enabled
 from transformers.modeling_outputs import (
     BaseModelOutput,
     BaseModelOutputWithPastAndCrossAttentions,
@@ -34,7 +34,7 @@ from transformers.modeling_outputs import (
 from transformers.utils import logging
 from transformers.modeling_utils import PreTrainedModel
-from .configuration_indictrans import IndicTransConfig
 logger = logging.get_logger(__name__)

 from torch.nn import functional as F
 from transformers.activations import ACT2FN
+from transformers.integrations.deepspeed import is_deepspeed_zero3_enabled
 from transformers.modeling_outputs import (
     BaseModelOutput,
     BaseModelOutputWithPastAndCrossAttentions,
 from transformers.utils import logging
 from transformers.modeling_utils import PreTrainedModel
+from configuration_indictrans import IndicTransConfig
 logger = logging.get_logger(__name__)