VatsaDev / Samplers

Implementing sampling methods and features from HF transformers/llama.cpp/kobold/etc into Nano-GPT

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Samplers

Implementing sampling methods and features from HF-transformers/llama.cpp/kobold/etc into Nano-GPT

Implemented:

  • Top K -> Being able to pick the top K tokens, Set k=1 for greedy, otherwise its multinomial
  • Top P -> Unuseable, need to fix the summation

About

Implementing sampling methods and features from HF transformers/llama.cpp/kobold/etc into Nano-GPT

License:MIT License


Languages

Language:Jupyter Notebook 93.3%Language:Python 6.7%