josh 5 月之前
父节点
当前提交
3a1871c84b
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      scripts/build_exo.py

+ 1 - 1
scripts/build_exo.py

@@ -33,7 +33,7 @@ def run():
             "--nofollow-import-to=tinygrad"
         ])
         inference_modules = [
-            name for _, name, _ in pkgutil.iter_modules(['exo/inference/models'])
+            name for _, name, _ in pkgutil.iter_modules(['exo/inference/mlx/models'])
         ]
         for module in inference_modules:
             command.append(f"--include-module=exo.inference.mlx.models.{module}")