Skip to content

Commit ac1560f

Browse files
committed
Update on "CI test for lora"
Add CI test for LoRA: note, not program-data separated yet. Differential Revision: [D78751767](https://our.internmc.facebook.com/intern/diff/D78751767/) [ghstack-poisoned]
2 parents 684858c + 682d600 commit ac1560f

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

.github/workflows/pull.yml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -684,6 +684,9 @@ jobs:
684684
# install llama requirements
685685
bash examples/models/llama/install_requirements.sh
686686
687+
# install a recent version of torchtune.
688+
PYTHON_EXECUTABLE=python python -m pip install torchtune==0.7.0.dev20250730 --extra-index-url https://download.pytorch.org/whl/nightly/cpu
689+
687690
# run llama runner in eager mode
688691
PYTHON_EXECUTABLE=python bash .ci/scripts/test_llama_runner_eager.sh
689692

requirements-examples.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,5 +3,5 @@
33
datasets == 3.6.0 # 4.0.0 deprecates trust_remote_code and load scripts. For now pin to 3.6.0
44
timm == 1.0.7
55
torchsr == 1.0.4
6-
torchtune >= 0.7.0
6+
torchtune >= 0.6.1
77
transformers == 4.53.1

0 commit comments

Comments
 (0)