Skip to content

Commit 5050c0c

Browse files
committed
drop deepspeed
1 parent 747bb19 commit 5050c0c

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

.lightning/workflows/fabric.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,9 +46,10 @@ run: |
4646
4747
if [ "${TORCH_VER}" == "2.1" ]; then
4848
echo "Set oldest versions"
49-
cd requirements/fabric
5049
pip install -U "lightning-utilities[cli]"
50+
cd requirements/fabric
5151
python -m lightning_utilities.cli requirements set-oldest --req_files "['base.txt', 'strategies.txt']"
52+
python -m lightning_utilities.cli requirements prune-pkgs --packages deepspeed --req_files strategies.txt
5253
cd ../..
5354
pip install "cython<3.0" wheel # for compatibility
5455
fi

.lightning/workflows/pytorch.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,9 +46,10 @@ run: |
4646
4747
if [ "${TORCH_VER}" == "2.1" ]; then
4848
recho "Set oldest versions"
49-
cd requirements/pytorch
5049
pip install -U "lightning-utilities[cli]"
50+
cd requirements/pytorch
5151
python -m lightning_utilities.cli requirements set-oldest --req_files "['base.txt', 'extra.txt', 'strategies.txt', 'examples.txt']"
52+
python -m lightning_utilities.cli requirements prune-pkgs --packages deepspeed --req_files strategies.txt
5253
cd ../..
5354
pip install "cython<3.0" wheel # for compatibility
5455
fi

0 commit comments

Comments
 (0)