Gemma / app.py
ShubhamD95's picture
Gemma.py
98a1c89 verified
raw
history blame
209 Bytes
from transformers import AutoTokenizer, AutoModelForCausalLM
model_name = "google/gemma-3-1b-it"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)