Downloads · 30 days
7
12% of all-time downloads
kanye-west-gpt2
kkenna/kanye-west-gpt2
kanye-west-gpt2 is a text generation model from kkenna. Use it when you need the model to write or continue text. It is set up for peft. The card lists the license as mit.
Fine-tuned GPT-2 on Kanye West lyrics using LoRA.
All-time downloads
60
Public
Repo size
6.5 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors6.5 MB · 66%
At a glance
- Task
- Text Generation
- Library
- peft
- License
- mit
- Access
- Public
- Created
- Jan 24, 2026
- Updated
- Jan 24, 2026
- SHA
- 6ffc4817
Try a prompt
Base models
Datasets
Model card
From the Hugging Face model README
kanye-west-gpt2
Fine-tuned GPT-2 on Kanye West lyrics using LoRA.
Usage
from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel
# Load
base_model = AutoModelForCausalLM.from_pretrained("gpt2")
model = PeftModel.from_pretrained(base_model, "kkenna/kanye-west-gpt2")
tokenizer = AutoTokenizer.from_pretrained("kkenna/kanye-west-gpt2")
# Generate
prompt = "<|lyrics|>\n"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=100, temperature=0.8, do_sample=True)
print(tokenizer.decode(outputs[0]))
Training
- Base: gpt2
- Method: LoRA (r=16, alpha=32)
- Epochs: 3
- Learning Rate: 0.0002
Disclaimer
For educational/research purposes. Song lyrics are copyrighted.
- Task
- Text Generation
- Library
- peft
- License
- mit
- Created
- Jan 24, 2026
- Updated
- Jan 24, 2026