josh 5 months ago
parent
commit
3a1871c84b
1 changed files with 1 additions and 1 deletions
  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}")