From d054a160db9a4d14b770c48917a4007a03ed0782 Mon Sep 17 00:00:00 2001 From: Wang Wei Date: Tue, 25 Mar 2025 19:13:34 +0800 Subject: [PATCH 1/2] Update pyproject.toml --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 099564c7..d7e8b3b1 100755 --- a/pyproject.toml +++ b/pyproject.toml @@ -34,7 +34,7 @@ vila-infer = "llava.cli.infer:main" vila-upload = "llava.cli.upload2hf:main" [project.optional-dependencies] -train = ["deepspeed==0.9.5", "ninja", "wandb"] +train = ["deepspeed>=0.9.5", "ninja", "wandb"] eval = ["word2number", "Levenshtein", "nltk", "pywsd"] [project.urls] From fbdcbe72460b8e203b20a9d0373ed5c506e9fcc8 Mon Sep 17 00:00:00 2001 From: Wang Wei Date: Tue, 25 Mar 2025 19:14:12 +0800 Subject: [PATCH 2/2] Update environment_setup.sh --- environment_setup.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/environment_setup.sh b/environment_setup.sh index c0a79012..ef748e27 100755 --- a/environment_setup.sh +++ b/environment_setup.sh @@ -21,7 +21,7 @@ pip install --upgrade pip setuptools pip install https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.8/flash_attn-2.5.8+cu122torch2.3cxx11abiFALSE-cp310-cp310-linux_x86_64.whl # Install VILA -pip install -e ".[train,eval]" +pip install -e ".[train,eval]" --use-pep517 # Quantization requires the newest triton version, and introduce dependency issue pip install triton==3.1.0