From ef5bbb9f67158818d09ad818bc017e8442a230a5 Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Sun, 8 Dec 2024 23:57:51 -0500 Subject: [PATCH] Fix install bug --- install.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/install.py b/install.py index b47188c..e2d51a4 100644 --- a/install.py +++ b/install.py @@ -59,7 +59,7 @@ def install_fa2(compile=False, ci=False): env = os.environ.copy() # limit max jobs to save memory in CI if ci: - env["MAX_JOBS"] = 4 + env["MAX_JOBS"] = "4" FA2_PATH = REPO_PATH.joinpath("submodules", "flash-attention") cmd = [sys.executable, "setup.py", "install"] subprocess.check_call(cmd, cwd=str(FA2_PATH.resolve()), env=env) @@ -74,7 +74,7 @@ def install_fa3(ci=False): env = os.environ.copy() # limit max jobs to save memory in CI if ci: - env["MAX_JOBS"] = 4 + env["MAX_JOBS"] = "4" cmd = [sys.executable, "setup.py", "install"] subprocess.check_call(cmd, cwd=str(FA3_PATH.resolve()), env=env)