Skip to content

Commit

Permalink
fix custom dataset test
Browse files Browse the repository at this point in the history
  • Loading branch information
IgorKasianenko committed Jan 15, 2025
1 parent 8336aac commit 380d0fa
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/llama_cookbook/configs/datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ class alpaca_dataset:
@dataclass
class custom_dataset:
dataset: str = "custom_dataset"
file: str = "recipes/quickstart/finetuning/datasets/custom_dataset.py"
file: str = "src/llama_cookbook/datasets/custom_dataset.py"
train_split: str = "train"
test_split: str = "validation"
data_path: str = ""
Expand Down

0 comments on commit 380d0fa

Please sign in to comment.