Browse Source

修复路径问题

lxylxy123321 1 week ago
parent
commit
d972319f29
1 changed files with 3 additions and 2 deletions
  1. 3 2
      backend/app/services/model_test_service.py

+ 3 - 2
backend/app/services/model_test_service.py

@@ -61,9 +61,10 @@ print(json.dumps({'generated_text': gen}))
     python = settings.compute_node_python
 
     # 使用 docker exec -i + heredoc 传递脚本到容器内 Python stdin,
-    # 避免长命令被截断或引号解析错误
+    # 避免长命令被截断或引号解析错误;-w 指定工作目录确保 app 模块可导入
     remote_cmd = (
-        f"docker exec -i {container} {python} << 'PYTHON_SCRIPT_EOF'\n"
+        f"docker exec -i -w {settings.compute_node_workdir} {container} "
+        f"{python} << 'PYTHON_SCRIPT_EOF'\n"
         f"{python_script}\n"
         f"PYTHON_SCRIPT_EOF"
     )