mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-06-24 17:31:18 +08:00
13 lines
316 B
Markdown
13 lines
316 B
Markdown
|
# Transformer LM
|
||
|
|
|
||
|
|
This is an example of a decoder-only Transformer LM. The only dependency is
|
||
|
|
MLX.
|
||
|
|
|
||
|
|
Run the example on the GPU with:
|
||
|
|
|
||
|
|
```
|
||
|
|
python main.py --gpu
|
||
|
|
```
|
||
|
|
|
||
|
|
By default the dataset is the [PTB corpus](https://paperswithcode.com/dataset/penn-treebank). Choose a different dataset with the `--dataset` option.
|