mirror of
https://github.com/karpathy/nanochat.git
synced 2026-04-06 07:35:32 +00:00
Add is_mps_device() and should_use_torch_compile() to nanochat/common.py Disable torch.compile on macOS MPS devices (prevents indefinite hanging) Add conditional torch.compile in base_train.py and chat_sft.py Add memory monitoring with 32GB inference / 96GB training limits Reference: Task-20, Task-18, Task-19, Task-28, Task-39 |
||
|---|---|---|
| .. | ||
| base_eval.py | ||
| base_train.py | ||
| chat_cli.py | ||
| chat_eval.py | ||
| chat_rl.py | ||
| chat_sft.py | ||
| chat_web.py | ||
| tok_eval.py | ||
| tok_train.py | ||