codeload/

Gemmini

from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b") model = AutoModelForCausalLM.from_pretrained("google/gemma-7b")

input_text = "Write me a poem about Machine Learning." input_ids = tokenizer(input_text, return_tensors="pt")

outputs = model.generate(**input_ids) print(tokenizer.decode(outputs[0]))