tacit/nanochat
1
0
Fork 0
mirror of https://github.com/karpathy/nanochat.git synced 2026-03-21 20:33:23 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

remove unused import

Browse Source
This commit is contained in:
Dipesh Babu 2026-02-09 20:51:15 -05:00
parent 3cb530cf97
commit 5c5cff25a5
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nanochat/gpt.py
View File

@ -12,7 +12,6 @@ Notable features:
- Flash Attention 3 integration
"""
from functools import partial
from dataclasses import dataclass
import torch
@ -22,7 +21,6 @@ import torch.nn.functional as F
from nanochat.common import get_dist_info, print0
from nanochat.optim import MuonAdamW, DistMuonAdamW
from typing import Optional
# Our custom Flash Attention module that automatically uses FA3 on Hopper+ and SDPA fallback elsewhere
from nanochat.flash_attention import flash_attn