indigo.tf / README.md
adyoi's picture
Upload folder using huggingface_hub
47dfda4 verified
|
Raw
History Blame Contribute Delete
872 Bytes
metadata
language:
  - id
license: mit
tags:
  - text-generation
  - from-scratch
  - gpt
  - transformer
  - tensorflow
  - keras

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.