Sfoglia il codice sorgente

修复远程引擎问题

lxylxy123321 1 settimana fa
parent
commit
49b2e9352e

+ 16 - 4
backend/app/engines/multimodal_engine.py

@@ -2,11 +2,23 @@ import json
 from pathlib import Path
 from typing import Any
 
-from app.config import get_settings
-from app.core.logging import logger
-from app.engines.base import BaseEngine
+# 远程训练节点可能没有 pydantic-settings,用环境变量兜底
+try:
+    from app.config import get_settings
+    settings = get_settings()
+except ImportError:
+    from types import SimpleNamespace
+    settings = SimpleNamespace(
+        data_dir=Path("/root/Fine-tuning/backend/data"),
+        processed_dir=Path("/root/Fine-tuning/backend/data") / "processed",
+        adapters_dir=Path("/root/Fine-tuning/backend/data") / "adapters",
+        models_dir=Path("/root/Fine-tuning/backend/data") / "models",
+    )
+
+import logging
+logger = logging.getLogger(__name__)
 
-settings = get_settings()
+from app.engines.base import BaseEngine
 
 
 class MultimodalEngine(BaseEngine):

+ 16 - 4
backend/app/engines/vision_engine.py

@@ -2,11 +2,23 @@ import json
 from pathlib import Path
 from typing import Any
 
-from app.config import get_settings
-from app.core.logging import logger
-from app.engines.base import BaseEngine
+# 远程训练节点可能没有 pydantic-settings,用环境变量兜底
+try:
+    from app.config import get_settings
+    settings = get_settings()
+except ImportError:
+    from types import SimpleNamespace
+    settings = SimpleNamespace(
+        data_dir=Path("/root/Fine-tuning/backend/data"),
+        processed_dir=Path("/root/Fine-tuning/backend/data") / "processed",
+        adapters_dir=Path("/root/Fine-tuning/backend/data") / "adapters",
+        models_dir=Path("/root/Fine-tuning/backend/data") / "models",
+    )
+
+import logging
+logger = logging.getLogger(__name__)
 
-settings = get_settings()
+from app.engines.base import BaseEngine
 
 
 class VisionEngine(BaseEngine):