Views
No views yet
google/gemma-4-31B-it, fine-tuned for SEC filing analysis on TPU v6e-8 with PyTorch/XLA SPMD FSDPv2.| Model | BERTScore F1 | Δ vs base |
|---|---|---|
Base gemma-4-31B-it | 0.8283 | — |
| + this adapter | 0.8760 | +5.76% |
q/k/v/o/gate/up/down_proj (regex-scoped to .language_model. for multimodal Gemma 4)(8, 1) over ("fsdp", "tensor")1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3import torch
4
5base = AutoModelForCausalLM.from_pretrained(
6 "google/gemma-4-31B-it",
7 torch_dtype=torch.bfloat16,
8 device_map="auto",
9)
10tok = AutoTokenizer.from_pretrained("google/gemma-4-31B-it")
11model = PeftModel.from_pretrained(base, "Srx7703/gemma-4-31b-financial-adapter")
12
13prompt = "What are the principal risk factors disclosed in NVIDIA's most recent 10-K?"
14inputs = tok.apply_chat_template(
15 [{"role": "user", "content": prompt}],
16 add_generation_prompt=True,
17 return_tensors="pt",
18).to(model.device)
19out = model.generate(inputs, max_new_tokens=256)
20print(tok.decode(out[0][inputs.shape[1]:], skip_special_tokens=True))