Gemma 2 9B Code Commentator

Published by @autodoc_ai · Community adapter

Generates docstrings, JSDoc, Sphinx, and Javadoc comments from function signatures and bodies. Infers parameter types, return values, and notable side effects.

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-code-comment-lora")

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

• transformers>=4.42