changed the path

modeling_ablang2paired.py

@@ -2,8 +2,16 @@ import torch
 import os
 from torch import nn
 from transformers import PreTrainedModel
-from ablang2.models.ablang2.ablang import AbLang as AbLang2
-from ablang2.configuration_ablang2paired import AbLang2PairedConfig
+
+# Import logic to handle the nested directory structure
+try:
+    # Try the relative import first (for Hugging Face Hub loading)
+    from .ablang2.models.ablang2.ablang import AbLang as AbLang2
+    from .ablang2.configuration_ablang2paired import AbLang2PairedConfig
+except ImportError:
+    # Fallback for local development
+    from ablang2.models.ablang2.ablang import AbLang as AbLang2
+    from ablang2.configuration_ablang2paired import AbLang2PairedConfig
 
 class AbLang2PairedHFModel(PreTrainedModel):
     config_class = AbLang2PairedConfig