Gemma-2-9B Email Writer
Official adapter · maintained by ModelForgeLab
Business email composition with tone control. Supports formal, neutral, and friendly registers. Trained on curated professional correspondence.
Usage
from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("google/gemma-2-9b-it")
base = AutoModelForCausalLM.from_pretrained("google/gemma-2-9b-it")
model = PeftModel.from_pretrained(base, "modelforgelab/gemma2-9b-email-writer")
inputs = tokenizer("Your prompt here", return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=256)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
Compatibility
- peft>=0.11
- transformers>=4.44