Views
No views yet
ananddey/akhorika-e2b-base)Gemma4ForConditionalGeneration (Google Gemma 4 E2B)as), English (en)1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_id = "ananddey/akhorika-e2b-base"
5
6tokenizer = AutoTokenizer.from_pretrained(model_id)
7model = AutoModelForCausalLM.from_pretrained(
8 model_id,
9 torch_dtype=torch.bfloat16,
10 device_map="auto",
11 attn_implementation="sdpa"
12)
13
14prompt = "অসম ভাৰতৰ উত্তৰ-পূব অঞ্চলৰ"
15inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
16
17with torch.inference_mode():
18 outputs = model.generate(
19 **inputs,
20 max_new_tokens=100,
21 temperature=0.7,
22 top_p=0.9,
23 repetition_penalty=1.1,
24 pad_token_id=0,
25 eos_token_id=2
26 )
27
28print(tokenizer.decode(outputs[0], skip_special_tokens=True))