This commit is contained in:
josh
2024-11-19 08:00:41 -08:00
parent 97ed990a98
commit 3a1871c84b

View File

@@ -33,7 +33,7 @@ def run():
"--nofollow-import-to=tinygrad" "--nofollow-import-to=tinygrad"
]) ])
inference_modules = [ 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: for module in inference_modules:
command.append(f"--include-module=exo.inference.mlx.models.{module}") command.append(f"--include-module=exo.inference.mlx.models.{module}")