# nanoGPT Shakespeare Model

このリポジトリには、シェイクスピアデータセットで事前学習した nanoGPT モデル (ckpt.pt) と推論コード (sample.py) が含まれています。

nanoGPT は Andrej Karpathy による GPT アーキテクチャのミニマリスト実装で、教育目的で設計されています。小規模な言語モデルのトレーニングとサンプリングを簡単に実現します。

https://github.com/karpathy/nanoGPT のリポジトリに従って、シェイクスピアの文字レベルモデルを Mac Studio でトレーニングしました。パラメータ数は約10Mです。

## 使い方
PyTorch でモデルをロードし、sample.py を使用してテキスト生成を行います。

例:
```python
python sample.py --out_dir=out-shakespeare-char --device=mps
```