/BabyGPT

notebook for basic text generation with Transformers

Primary LanguageJupyter Notebook

BabyGPT

Notebooks for basic text generation with Transformers

  • BabyGPT_PyTorch_Only.ipynb - Basic GPT class built using Simple Transformer coded in PyTorch

  • BabyGPT_PyTorch_Tranformer.ipynb - Basic GPT class built using TransformerEncoder from torch.nn

  • HF_Transformer_infer.ipynb - Direct Inference from Hugging Face transformers