from transformers import Gemma3nForConditionalGeneration
import torch

model_id = "google/gemma-3n-e4b-it"

model = Gemma3nForConditionalGeneration.from_pretrained(model_id, device_map="auto", dtype=torch.bfloat16).eval()
torch.save(model.state_dict(), f"{model_id.split("/")[1]}.bin")