Views
No views yet
1from transformers import AutoProcessor, AutoModelForImageTextToText
2import torch
3
4model = AutoModelForImageTextToText.from_pretrained(
5 "supersamdev/gemma-4-e4b-dailydialog-16bit",
6 torch_dtype=torch.bfloat16,
7 device_map="auto",
8)
9processor = AutoProcessor.from_pretrained("supersamdev/gemma-4-e4b-dailydialog-16bit")
10
11history = [{"role": "user", "content": [{"type": "text", "text": "Hey, how was your day?"}]}]
12inputs = processor.apply_chat_template(history, add_generation_prompt=True, tokenize=True, return_tensors="pt").to(model.device)
13input_len = inputs["input_ids"].shape[1]
14
15outputs = model.generate(**inputs, max_new_tokens=150, temperature=0.7, do_sample=True, repetition_penalty=1.1)
16print(processor.decode(outputs[0][input_len:], skip_special_tokens=True))Gemma is provided under and subject to the Gemma Terms of Use found at ai.google.dev/gemma/terms