Remove remaining ablang2 package import from modeling_ablang2paired.py

modeling_ablang2paired.py

@@ -9,33 +9,29 @@ try:
 except ImportError:
     from configuration_ablang2paired import AbLang2PairedConfig
 
-# Import the AbLang model from the integrated files
-try:
-    from ablang2.models.ablang2.ablang import AbLang
-except ImportError:
-    # Fallback: try to find the ablang.py file in the current directory structure
-    import importlib.util
-    import os
-    
-    def load_ablang_module():
-        """Load the AbLang module from the nested directory structure."""
-        # Try to find the ablang.py file in the nested directory
-        current_dir = os.path.dirname(os.path.abspath(__file__))
-        ablang_path = os.path.join(current_dir, "ablang2", "models", "ablang2", "ablang.py")
-        
-        if os.path.exists(ablang_path):
-            spec = importlib.util.spec_from_file_location("ablang", ablang_path)
-            ablang_module = importlib.util.module_from_spec(spec)
-            spec.loader.exec_module(ablang_module)
-            return ablang_module.AbLang
-        else:
-            # If not found, raise an error with helpful message
-            raise ImportError(
-                "Could not find AbLang module. Please ensure the ablang2 directory structure is present "
-                "or install the ablang2 package."
-            )
+# Import the AbLang model from the local file structure
+import importlib.util
+import os
+
+def load_ablang_module():
+    """Load the AbLang module from the local directory structure."""
+    # Try to find the ablang.py file in the local directory
+    current_dir = os.path.dirname(os.path.abspath(__file__))
+    ablang_path = os.path.join(current_dir, "ablang2", "models", "ablang2", "ablang.py")
     
-    AbLang = load_ablang_module()
+    if os.path.exists(ablang_path):
+        spec = importlib.util.spec_from_file_location("ablang", ablang_path)
+        ablang_module = importlib.util.module_from_spec(spec)
+        spec.loader.exec_module(ablang_module)
+        return ablang_module.AbLang
+    else:
+        # If not found, raise an error with helpful message
+        raise ImportError(
+            "Could not find AbLang module. Please ensure the ablang2 directory structure is present "
+            "in the repository."
+        )
+
+AbLang = load_ablang_module()
 
 class AbLang2PairedHFModel(PreTrainedModel):
     config_class = AbLang2PairedConfig