Skip to content

Commit bc8d6ac

Browse files
[fine_tuning]: change test settings
1 parent f3d9a73 commit bc8d6ac

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

projects/fine_tuning/testing/config.yaml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -93,9 +93,9 @@ ci_presets:
9393
tests.fine_tuning.matbenchmarking.stop_on_error: false
9494

9595
tests.fine_tuning.test_settings.dataset_name: alpaca_data.json
96-
tests.fine_tuning.test_settings.dataset_replication: 0.5
96+
tests.fine_tuning.test_settings.dataset_replication: 0.2
9797

98-
tests.fine_tuning.test_settings.gpu: 8
98+
tests.fine_tuning.test_settings.gpu: [2, 4]
9999

100100
tests.fine_tuning.test_settings.hyper_parameters.gradient_accumulation_steps: 1
101101
tests.fine_tuning.test_settings.hyper_parameters.per_device_train_batch_size: 4
@@ -105,9 +105,9 @@ ci_presets:
105105

106106
tests.fine_tuning.test_extra_settings:
107107
- model_name:
108-
- meta-llama/Llama-2-13b-hf
108+
# - meta-llama/Llama-2-13b-hf
109109
# - meta-llama/Meta-Llama-3.1-70B
110-
# - ibm-granite/granite-3b-code-instruct
110+
- ibm-granite/granite-3b-code-instruct
111111
# - instructlab/granite-7b-lab
112112
# - ibm-granite/granite-8b-code-base
113113
# - meta-llama/Meta-Llama-3.1-8B

0 commit comments

Comments
 (0)