DualBrain / llama2.cs_trrahul

Inference Llama 2 in one file of pure C#

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Andrej Karpathy's llama2.c in one file of pure C#.

llama2.c is a very simple implementation to run inference of models with a Llama2-like transformer-based LLM architecture.

This is a pure C# implementation of the same thing. It is optimized for speed and very simple to understand and modify.

Usage

Requires .net7 or higher.

  1. First put the stories15M.bin file in the same directory as the executable. You can download it from here
  2. Get tokenizer from here and put it in the same directory as the executable.
dotnet build -c Release

Generate a random story

.\bin\Release\net7.0\llama2.cs.exe stories15M.bin

WindowsTerminal_LrRTW3joph

Generate a random story with a given prompt

.\bin\Release\net7.0\llama2.cs.exe stories15M.bin  -i "A long time ago a"

TODO

  • Inference with Llama2 checkpoints
  • Use high performance C# types from .net8?
  • Add training functionality

About

Inference Llama 2 in one file of pure C#

License:MIT License


Languages

Language:C# 100.0%