chore: change cache dir in configs
Browse files
configs/model/distill_llama3_1_8b_lk_smd_wtk64_fd64_w01.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1 |
name: llama
|
2 |
model:
|
3 |
pretrained_model_name_or_path: "meta-llama/Meta-Llama-3.1-8B"
|
4 |
-
cache_dir: "
|
5 |
return_dict: true
|
6 |
load_in_8bit: false
|
7 |
load_in_4bit: false
|
|
|
1 |
name: llama
|
2 |
model:
|
3 |
pretrained_model_name_or_path: "meta-llama/Meta-Llama-3.1-8B"
|
4 |
+
cache_dir: "." # Set this to where you want to save checkpoint weights
|
5 |
return_dict: true
|
6 |
load_in_8bit: false
|
7 |
load_in_4bit: false
|