zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-01 12:49:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

black format

Browse Source
This commit is contained in:
Awni Hannun
2023-12-09 14:15:25 -08:00
parent b8332a1e66
commit 98f4346c81
6 changed files with 44 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
transformer_lm/datasets.py
View File

@@ -42,7 +42,7 @@ def wikitext(dataset="2", save_dir="/tmp"):
Load the WikiText-* language modeling dataset:
https://paperswithcode.com/dataset/wikitext-2
https://paperswithcode.com/dataset/wikitext-103
"""
if dataset not in ("2", "103"):
raise ValueError(f'Dataset must be either "2" or "103", got {dataset}')