CoolFace
Modelpublic

mayankchugh-learning/gpt2-lora-ai-demo

sourceHugging Facemitupdated 3mo agoView on Hugging Face
0likes5downloads
Model Card

GPT-2 + LoRA Fine-Tuned (AI Demo)

Fine-tuned with PEFT/LoRA on a small AI Q&A corpus.

Usage

python
from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer

base  = AutoModelForCausalLM.from_pretrained("gpt2")
model = PeftModel.from_pretrained(base, "mayankchugh-learning/gpt2-lora-ai-demo")
tok   = AutoTokenizer.from_pretrained("mayankchugh-learning/gpt2-lora-ai-demo")

inputs = tok("Hugging Face is", return_tensors="pt")
output = model.generate(**inputs, max_new_tokens=40)
print(tok.decode(output[0]))

Training