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

You might also like