Skip to content

Commit a5a49f5

Browse files
javierdlrmmanu-sj
authored andcommitted
[HWORKS-2040] Add lora fine-tuning tutorial (#303)
1 parent 24d188a commit a5a49f5

File tree

5 files changed

+658
-342
lines changed

5 files changed

+658
-342
lines changed

llm-ai-systems/fine-tuning/configs/lora_configs/lora.json

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
"r": 8,
33
"lora_alpha": 16,
44
"lora_dropout": 0.05,
5-
"target_modules": ["q_proj", "v_proj", "k_proj", "o_proj", "gate_proj", "up_proj", "down_proj", "embed_tokens", "lm_head"],
5+
"target_modules": ["gate_proj", "up_proj", "down_proj"],
66
"task_type": "CAUSAL_LM",
77
"modules_to_save": [],
88
"bias": "none",

llm-ai-systems/fine-tuning/llama_fine_tune_runtime_env.yaml

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@ pip:
66
env_vars:
77
LIBRARY_PATH: "$CUDA_HOME/lib64:$LIBRARY_PATH"
88
PROJECT_DIR: "/home/yarnapp/hopsfs"
9-
TRAINED_MODEL_STORAGE_PATH: "${PROJECT_DIR}/Resources/llama_finetuning/fine-tuned-model"
10-
TRAINING_DATA_DIR: "${PROJECT_DIR}/Resources/llama_finetuning/datasets"
11-
TRAINING_CONFIGURATION_DIR: "${PROJECT_DIR}/Resources/llama_finetuning/configs"
12-
9+
TRAINING_DATA_DIR: "${PROJECT_DIR}/Resources/llama_finetuning/datasets" # hopsfs path to training data files
10+
TRAINING_CONFIGURATION_DIR: "${PROJECT_DIR}/Resources/llama_finetuning/configs" # hopsfs path to training config files
11+
TRAINING_CHECKPOINTS_DIR: "${PROJECT_DIR}/Resources/llama_finetuning/checkpoints" # hopsfs path to torch trainer checkpoints

0 commit comments

Comments
 (0)