code-gemma / README.md
gnumanth's picture
chore: more info
15b7dc1 verified
|
raw
history blame
620 Bytes
metadata
library_name: transformers
license: mit
language:
  - en
widget:
  - text: Reverse a string 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