Fix inference code: configuration_logos.py

configuration_logos.py

@@ -2,17 +2,11 @@
 
 from __future__ import annotations
 
-import sys
-import importlib
-from pathlib import Path
+from types import SimpleNamespace
 from typing import Any, Dict
 
 from transformers import PretrainedConfig
 
-_MODULE_DIR = Path(__file__).resolve().parent
-if str(_MODULE_DIR) not in sys.path:
-    sys.path.insert(0, str(_MODULE_DIR))
-
 
 _NATIVE_DEFAULTS: Dict[str, Any] = {
     "vocab_size": 32000,
@@ -100,15 +94,13 @@ class LogosConfig(PretrainedConfig):
         }
 
     def to_native_config(self):
-        model_dir = getattr(self, "_name_or_path", None) or getattr(self, "name_or_path", None)
-        if model_dir:
-            model_path = Path(str(model_dir)).resolve()
-            if model_path.exists() and str(model_path) not in sys.path:
-                sys.path.insert(0, str(model_path))
-        NativeLogosConfig = importlib.import_module("models.logos").LogosConfig
-
         data = {name: getattr(self, name) for name in _NATIVE_DEFAULTS}
-        return NativeLogosConfig(**data)
+        data["num_layers"] = (
+            int(data["num_entry_layers"])
+            + int(data["num_body_layers"])
+            + int(data["num_exit_layers"])
+        )
+        return SimpleNamespace(**data)
 
 
 __all__ = ["LogosConfig"]