code-gemma / README.md
gnumanth's picture
chore: format
2fb3bc7 verified
metadata
library_name: transformers
license: mit
language:
  - en
widget:
  - text: Give me code to reverse a string in python
  - text: How do I find today's date in python?

code-gemma

Google's gemma-2b-it trained code_instructions_122k_alpaca_style dataset

Usage

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-generation", model="gnumanth/code-gemma")
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("gnumanth/code-gemma")
model = AutoModelForCausalLM.from_pretrained("gnumanth/code-gemma")

Hemanth HM