Skip to content

Commit 3406c0b

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 8ebeba2 + 006d18d commit 3406c0b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.ci/scripts/test_llama_lora.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ DOWNLOADED_PATH=$(
4848
--model_id "${HF_MODEL_REPO}" \
4949
--files "adapter_config.json" "adapter_model.pt" "consolidated.00.pth" "params.json" "tokenizer.model"
5050
)
51-
51+
EXPORTED_MODEL_NAME="llama_3_2_1B_lora.pte"
5252
# Export model.
5353
$PYTHON_EXECUTABLE -m extension.llm.export.export_llm \
5454
base.checkpoint="${DOWNLOADED_PATH}/consolidated.00.pth" \
@@ -61,7 +61,7 @@ $PYTHON_EXECUTABLE -m extension.llm.export.export_llm \
6161
model.dtype_override="fp32" \
6262
backend.xnnpack.enabled=true \
6363
backend.xnnpack.extended_ops=true \
64-
export.output_name="llama3_2_1B_lora.pte"
64+
export.output_name="${EXPORTED_MODEL_NAME}"
6565

6666
# Build llama runner.
6767
cmake_install_executorch_libraries

0 commit comments

Comments
 (0)