소스 검색

fix custom dataset test

Igor Kasianenko 4 달 전
부모
커밋
ec214112f3
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      src/llama_recipes/configs/datasets.py

+ 1 - 1
src/llama_recipes/configs/datasets.py

@@ -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 = ""