KV-cache quantization

Aperta
#20 0 commenti 0 reazioni 0 assegnatari Vedi su GitHub

Nessuno ha ancora preso questa issue.

Valutazione

Difficoltà
5/5
Tempo stimato
Più di una settimana
Idoneità per principianti
35/100
Tipo di issue
Funzionalità
Chiarezza
Abbastanza chiara
Stato di attività
Attiva
Stack tecnologico
huggingface, python, pytorch

Direzione di ricerca

Start by reviewing the Hugging Face model integration and existing quantization code to identify the cache interfaces and supported models. The issue calls for an ADR, a cache implementation with tests, and perplexity and memory measurements at 4k and 16k context for one model; those results define done. Expect substantial design and evaluation work.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Descrizione

area: formats enhancement help wanted

For long contexts the KV cache can be larger than the weights. Storing keys and values in
INT8 or FP8 roughly halves that memory.

Task: design and implement KV-cache quantization for Hugging Face models (for example a
custom Cache class that stores quantized keys and values per head with per-token
scales), and measure perplexity and memory at 4k and 16k context.

Done when there is a short design note (ADR), an implementation with tests, and
results for one model.

Lingua principale
Python
Stelle
0
Fork
0
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Preparare l'ambiente

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Altre issue di yashwanthdevavarapu/bitsmith

Tutte le issue di yashwanthdevavarapu/bitsmith

Issue simili

Altre issue su Python

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.