indigo.tf / README.md
adyoi's picture
Upload folder using huggingface_hub
47dfda4 verified
|
Raw
History Blame Contribute Delete
872 Bytes
---
language:
- id
license: mit
tags:
- text-generation
- from-scratch
- gpt
- transformer
- tensorflow
- keras
---
# Indigo-TF
Versi **TensorFlow/Keras** dari proyek [Indigo](https://huggingface.co/adyoi/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
```bash
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.