Views
No views yet
do_sample=False), 200 max new tokens.| Metric | FP16 (baseline) | Q8 | Q4 |
|---|---|---|---|
| SQNR | — | 27.49 dB | 18.75 dB |
| Top-1 Agreement | — | 92.9% | 81.1% |
| KL Divergence | — | 0.0496 | 0.3334 |
| Speed (tok/s) | 56.9 | 14.5 | 40.2 |
| VRAM | 9.5 GB | 7.4 GB | 6.3 GB |
| Category | SQNR | Top-1 Agreement | KL Divergence | Speed (tok/s) |
|---|---|---|---|---|
| 🔢 Math | 27.09 dB | 92.4% | 0.0424 | 14.8 |
| 🧠 Logic | 27.18 dB | 92.8% | 0.0802 | 13.9 |
| 💻 Code | 29.49 dB | 94.5% | 0.0346 | 14.8 |
| 🔬 Science | 26.34 dB | 92.1% | 0.0410 | 14.7 |
1from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
2import torch
3
4model = AutoModelForCausalLM.from_pretrained(
5 "MichaelLowrance/gemma-4-e2b-q8",
6 quantization_config=BitsAndBytesConfig(load_in_8bit=True),
7 device_map="cuda",
8)
9tokenizer = AutoTokenizer.from_pretrained("MichaelLowrance/gemma-4-e2b-q8")
10
11messages = [{"role": "user", "content": "Hello!"}]
12inputs = tokenizer.apply_chat_template(messages, return_tensors="pt", return_dict=True).to("cuda")
13
14with torch.no_grad():
15 output = model.generate(**inputs, max_new_tokens=200, do_sample=False)
16
17print(tokenizer.decode(output[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))