add missing fixture decorator for predownload dataset (#2117) [skip ci]

* add missing fixture decorator for predownload dataset

* also pre download the tokenizer files
This commit is contained in:
Wing Lian
2024-12-03 18:08:46 -05:00
committed by GitHub
parent d87df2c776
commit 418ad2b586
2 changed files with 13 additions and 0 deletions

View File

@@ -46,6 +46,7 @@ def download_mhenrichsen_alpaca_2k_w_revision_dataset():
)
@pytest.fixture(scope="session", autouse=True)
def download_mlabonne_finetome_100k_dataset():
# download the dataset
snapshot_download("mlabonne/FineTome-100k", repo_type="dataset")