model/__init__.py

from .config import Config
from .transformer_block import TransformerBlock
from .feed_forward import FeedForward
from .multi_head_attention import MultiHeadAttention
from .single_attention_head import AttentionHead
from .model import PotterGPT
from .tokenizer import CharacterLevelTokenizer