Downloads · 30 days
8
14% of all-time downloads
gpt2-cda-gender
ao9000/gpt2-cda-gender
gpt2-cda-gender is a text generation model from ao9000. Use it when you need the model to write or continue text. It is set up for transformers.
- Dataset: Wikipedia-10 - Target modules = ["query", "value"]
All-time downloads
56
Public
Parameters
124M
249 MB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors249 MB · 100%
At a glance
- Task
- Text Generation
- Library
- transformers
- Model type
- gpt2
- Access
- Public
- Created
- Jul 22, 2024
- Updated
- Jul 22, 2024
- SHA
- a875d867
Try a prompt
Model card
From the Hugging Face model README
Lora finetuning on Wikipedia-10, applying counter factual data augmentation (CDA)
- Dataset: Wikipedia-10
- Target modules = ["query", "value"]
{
"epoch": 0.06122556358609606,
"total_flos": 3.3909236563968e+16,
"train_loss": 0.9592373611629009,
"train_runtime": 5840.6966,
"train_samples": 1045315,
"train_samples_per_second": 10.958,
"train_steps_per_second": 0.342
}
Training script: https://github.com/ao9000/bias-bench/blob/main/experiments/run_clm.py
- Task
- Text Generation
- Library
- transformers
- Type
- gpt2
- Created
- Jul 22, 2024
- Updated
- Jul 22, 2024