fairseq2n source maybe?

facebookresearch · Dec 11, 2024 · 90d4a05 · 90d4a05
1 parent c3be1af
commit 90d4a05
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 8 deletions.
diff --git a/.github/workflows/lint_and_test.yml b/.github/workflows/lint_and_test.yml
@@ -28,7 +28,8 @@ jobs:
         run: |
           sudo apt-get install libsndfile1
           python -m pip install --upgrade pip
-          python -m pip install fairseq2==v0.3.0rc1 --pre --extra-index-url  https://fair.pkg.atmeta.com/fairseq2/whl/rc/pt2.5.1/cpu/ 
+          python -m pip install fairseq2n==0.3.0rc1+cpu --pre --extra-index-url  https://fair.pkg.atmeta.com/fairseq2/whl/rc/pt2.5.1/cpu/ 
+          python -m pip install fairseq2==0.3.0rc1 --pre --extra-index-url  https://fair.pkg.atmeta.com/fairseq2/whl/rc/pt2.5.1/cpu/ 
           python -m pip install -r requirements.txt
           python -m pip install -r requirements-dev.txt
           python -m pip install -e .
@@ -59,8 +60,9 @@ jobs:
       - name: Install dependencies
         run: |
           sudo apt-get install libsndfile1
-          python -m pip install --upgrade pip
-          python -m pip install fairseq2==v0.3.0rc1 --pre --extra-index-url  https://fair.pkg.atmeta.com/fairseq2/whl/rc/pt2.5.1/cpu/ 
+          python -m pip install --upgrade pip          
+          python -m pip install fairseq2n==0.3.0rc1+cpu --pre --extra-index-url  https://fair.pkg.atmeta.com/fairseq2/whl/rc/pt2.5.1/cpu/ 
+          python -m pip install fairseq2==0.3.0rc1 --pre --extra-index-url  https://fair.pkg.atmeta.com/fairseq2/whl/rc/pt2.5.1/cpu/ 
           python -m pip install -r requirements.txt
           python -m pip install -r requirements-dev.txt
           python -m pip install -e .
@@ -83,8 +85,9 @@ jobs:
           cache: "pip"
       - name: Install dependencies
         run: |
-          sudo apt-get install libsndfile1
-          python -m pip install fairseq2==v0.3.0rc1 --pre --extra-index-url  https://fair.pkg.atmeta.com/fairseq2/whl/rc/pt2.5.1/cpu/ 
+          sudo apt-get install libsndfile1          
+          python -m pip install fairseq2n==0.3.0rc1+cpu --pre --extra-index-url  https://fair.pkg.atmeta.com/fairseq2/whl/rc/pt2.5.1/cpu/ 
+          python -m pip install fairseq2==0.3.0rc1 --pre --extra-index-url  https://fair.pkg.atmeta.com/fairseq2/whl/rc/pt2.5.1/cpu/ 
           python -m pip install -r requirements.txt
           python -m pip install -r requirements-dev.txt
           python -m pip install -e .
@@ -109,7 +112,8 @@ jobs:
         run: |
           sudo apt-get install libsndfile1
           python -m pip install --upgrade pip
-          python -m pip install fairseq2==v0.3.0rc1 --pre --extra-index-url  https://fair.pkg.atmeta.com/fairseq2/whl/rc/pt2.5.1/cpu/ 
+          python -m pip install fairseq2n==0.3.0rc1+cpu --pre --extra-index-url  https://fair.pkg.atmeta.com/fairseq2/whl/rc/pt2.5.1/cpu/ 
+          python -m pip install fairseq2==0.3.0rc1 --pre --extra-index-url  https://fair.pkg.atmeta.com/fairseq2/whl/rc/pt2.5.1/cpu/ 
           python -m pip install -r requirements.txt
           python -m pip install -r requirements-dev.txt
           python -m pip install -e .

diff --git a/pyproject.toml b/pyproject.toml
@@ -31,7 +31,7 @@ classifiers = [
 ]
 
 dependencies = [
-  "fairseq2>=0.3.0rc1",
+#  "fairseq2>=0.3.0rc1", # install manually from https://fair.pkg.atmeta.com/fairseq2/whl/rc/pt2.5.1/cpu/
   "numpy>=1.21",
   "torch",
   "torchaudio",

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-fairseq2>=0.3.0rc1
+# fairseq2>=0.3.0rc1 # install manually from https://fair.pkg.atmeta.com/fairseq2/whl/rc/pt2.5.1/cpu/
 # See https://github.com/NVIDIA/TensorRT-LLM/issues/1145
 mpmath==1.3.0
 numpy>=1.21