Indigo-TF

Versi TensorFlow/Keras dari proyek Indigo — model bahasa GPT kecil dari nol. Repo utama (PyTorch): D:\Documents\GitHub\Indigo.

Format checkpoint identik (.safetensors + _meta.json, penanda backend: tensorflow), sehingga tidak bisa tertukar dengan checkpoint PyTorch.

Pakai

pip install -r requirements.txt

python train.py --data data/sample.txt --steps 2000
python generate.py --prompt "Indigo"

Fitur setara repo utama: pembersihan teks, split validasi per-file, best-checkpoint, resume (--init-from), tokenizer char/bpe, cosine LR + warmup, pilihan --device.

Catatan: GPU TensorFlow di Windows native tidak didukung sejak TF 2.11 — gunakan WSL2/Colab/Linux.

Downloads last month: -