bitext/Bitext-customer-support-llm-chatbot-training-dataset
Viewer • Updated • 26.9k • 5.65k • 186
How to use BOBsssss/gemma2-2b-customer-support-lora with PEFT:
Task type is invalid.
LoRA adapter fine-tuned from google/gemma-2-2b-it on the Bitext customer support dataset
(26k+ single-turn support instructions/responses, 11 categories / 27 intents).
Trained in fp16/bf16 (no quantization), loss masked to response tokens only.
from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel
base = AutoModelForCausalLM.from_pretrained("google/gemma-2-2b-it")
model = PeftModel.from_pretrained(base, "BOBsssss/gemma2-2b-customer-support-lora")
tokenizer = AutoTokenizer.from_pretrained("BOBsssss/gemma2-2b-customer-support-lora")
huggingface-cli download BOBsssss/gemma2-2b-customer-support-lora --local-dir ./adapter
ollama pull gemma2:2b
cat > Modelfile <<EOF
FROM gemma2:2b
ADAPTER ./adapter
EOF
ollama create support-bot -f Modelfile
ollama run support-bot "I need to cancel order 12345"