launcher_axmodel 移除对编译上下文的依赖

VideoX-Fun/examples/z_image_fun/launcher_axmodel.py

@@ -298,6 +298,7 @@ class AxSplitTransformer:
         if spec.label not in self._session_cache:
             path = self._expected_path(spec)
             self._session_cache[spec.label] = AxInferenceSession(path.as_posix())
+            logger.info(f"加载子图 session: {spec.label} from {path.name}")
         return self._session_cache[spec.label]
 
     def close(self) -> None:
@@ -333,33 +334,14 @@ class AxSplitTransformer:
     def _load_auto_specs(self) -> List[SubGraphSpec]:
         specs: List[SubGraphSpec] = []
         for path in sorted(self.model_dir.glob("auto_*.axmodel")):
-            # Ax compiler有两种输出目录命名：compiled_{stem} 或 compiled_auto_XX
-            candidates = [path.parent / f"compiled_{path.stem}"]
-            # 兼容实际产物 compiled_auto_00 这类名称
-            parts = path.stem.split("_", 3)
-            if len(parts) >= 2:
-                candidates.append(path.parent / f"compiled_{parts[0]}_{parts[1]}")
-
-            ctx_path = None
-            for cand in candidates:
-                candidate_ctx = cand / "build_context.json"
-                if candidate_ctx.exists():
-                    ctx_path = candidate_ctx
-                    break
-
-            if ctx_path is None:
-                logger.warning(f"跳过 {path.name}，缺少 build_context.json (尝试: {[c.as_posix() for c in candidates]})")
-                continue
-            if not ctx_path.exists():
-                logger.warning(f"跳过 {path.name}，缺少 build_context.json")
-                continue
             try:
-                with ctx_path.open("r", encoding="utf-8") as f:
-                    ctx = json.load(f)
-                inputs = [p.get("tensor_name") for p in ctx.get("cfg", {}).get("input_processors", []) if p.get("tensor_name")]
-                outputs = [p.get("tensor_name") for p in ctx.get("cfg", {}).get("output_processors", []) if p.get("tensor_name")]
+                session = AxInferenceSession(path.as_posix())
+                inputs = [info.name for info in session.get_inputs() if getattr(info, "name", None)]
+                outputs = [info.name for info in session.get_outputs() if getattr(info, "name", None)]
+                # 缓存 session，避免重复打开
+                self._session_cache[path.stem] = session
             except Exception as exc:  # pragma: no cover - defensive
-                logger.warning(f"解析 {ctx_path} 失败: {exc}")
+                logger.warning(f"跳过 {path.name}，加载/解析 IO 失败: {exc}")
                 continue
 
             if not inputs or not outputs: