Views
No views yet
google/gemma-4-E2B-it, trained for RepoMind call-chain analysis experiments.google/gemma-4-E2B-it and comply with the base model license/terms.google/gemma-4-E2B-itfull-synthetic-augmented-v2-2026062109601acfabebab623f543d76366b186f2da0bbc9143a69ffaff00e476802781aq_proj,k_proj,v_proj,o_proj,gate_proj,up_proj,down_projregex:.*(vision_tower|audio_tower).*8ef422a6eeb142d657da41deeb41abae9e8ca3e7f11f5fb1605b0b6af845b2c7| Step | Eval loss |
|---|---|
| initial | 2.231752 |
| 20 | 0.453445 |
| 40 | 0.300186 |
| 60 | 0.245931 |
| 80 | 0.206748 |
| 100 | 0.193902 |
no_overfit_signal_eval_loss_decreased.| Variant | Precision | Recall | Evidence accuracy |
|---|---|---|---|
| base | n/a | 0.000 | n/a |
| v1 adapter | 0.750 | 0.250 | 0.667 |
| v2 adapter | 0.750 | 0.250 | 0.333 |
1import torch
2from peft import AutoPeftModelForCausalLM
3from transformers import AutoTokenizer
4
5adapter_id = "Shirodora233/gemma4-e2b-repomind-qlora-synth-v2-pilot"
6
7tokenizer = AutoTokenizer.from_pretrained(adapter_id, trust_remote_code=True)
8model = AutoPeftModelForCausalLM.from_pretrained(
9 adapter_id,
10 torch_dtype=torch.bfloat16,
11 device_map="auto",
12 trust_remote_code=True,
13)
14
15messages = [
16 {
17 "role": "system",
18 "content": (
19 "Return repo-local symbol-level call edges as strict JSON. "
20 "Use fully qualified symbols, include file/line/evidence for every call edge."
21 ),
22 },
23 {
24 "role": "user",
25 "content": "{\"case_id\":\"example\",\"question\":\"Find repo-local call edges.\"}",
26 },
27]
28
29inputs = tokenizer.apply_chat_template(
30 messages,
31 tokenize=True,
32 add_generation_prompt=True,
33 return_tensors="pt",
34 return_dict=True,
35).to(model.device)
36
37with torch.inference_mode():
38 output_ids = model.generate(
39 **inputs,
40 max_new_tokens=768,
41 do_sample=False,
42 pad_token_id=tokenizer.eos_token_id,
43 )
44
45generated = output_ids[0, inputs["input_ids"].shape[-1]:]
46print(tokenizer.decode(generated, skip_special_tokens=True))adapter_model.safetensorsadapter_config.jsontokenizer.jsontokenizer_config.jsonchat_template.jinjametadata/training_summary.jsonmetadata/overfit_monitor.jsonmetadata/run_config.jsonmetadata/environment_snapshot.jsonmetadata/freeze-manifest.jsonmetadata/validation-summary.jsonreports/finetune-gemma4-e2b-qlora-frozen-synth-v2-100step-20260621.mdreports/finetune-gemma4-e2b-realcase-v1-v2-comparison-20260621.md