josh hai 5 meses
pai
achega
3a1871c84b
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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}")