|
@@ -51,7 +51,7 @@ def test_custom_dataset(step_lr, optimizer, get_model, tokenizer, train, mocker,
|
|
kwargs = {
|
|
kwargs = {
|
|
"dataset": "custom_dataset",
|
|
"dataset": "custom_dataset",
|
|
"model_name": llama_version,
|
|
"model_name": llama_version,
|
|
- "custom_dataset.file": "recipes/finetuning/datasets/custom_dataset.py",
|
|
|
|
|
|
+ "custom_dataset.file": "recipes/quickstart/finetuning/datasets/custom_dataset.py",
|
|
"custom_dataset.train_split": "validation",
|
|
"custom_dataset.train_split": "validation",
|
|
"batch_size_training": 2,
|
|
"batch_size_training": 2,
|
|
"val_batch_size": 4,
|
|
"val_batch_size": 4,
|
|
@@ -108,7 +108,7 @@ def test_unknown_dataset_error(step_lr, optimizer, tokenizer, get_model, train,
|
|
|
|
|
|
kwargs = {
|
|
kwargs = {
|
|
"dataset": "custom_dataset",
|
|
"dataset": "custom_dataset",
|
|
- "custom_dataset.file": "recipes/finetuning/datasets/custom_dataset.py:get_unknown_dataset",
|
|
|
|
|
|
+ "custom_dataset.file": "recipes/quickstart/finetuning/datasets/custom_dataset.py:get_unknown_dataset",
|
|
"batch_size_training": 1,
|
|
"batch_size_training": 1,
|
|
"use_peft": False,
|
|
"use_peft": False,
|
|
}
|
|
}
|