prajdabre
/

rotary-indictrans2-indic-en-1B

VarunGumma commited on Nov 1, 2024

Commit

0d08972

verified ·

1 Parent(s): 5f3ea6b

Update modeling_rotary_indictrans.py

Files changed (1) hide show

modeling_rotary_indictrans.py CHANGED Viewed

@@ -28,7 +28,7 @@ from torch.amp import autocast
 from torch import einsum
 from transformers.modeling_utils import PreTrainedModel
-from configuration_rotary_indictrans import RotaryIndicTransConfig
 from flash_attn import flash_attn_func, flash_attn_varlen_func
 from flash_attn.bert_padding import (

 from torch import einsum
 from transformers.modeling_utils import PreTrainedModel
+from .configuration_rotary_indictrans import RotaryIndicTransConfig
 from flash_attn import flash_attn_func, flash_attn_varlen_func
 from flash_attn.bert_padding import (