from src.model.gpt import GPTModel
from src.model.config import GPTConfig, GPT124M_CONFIG