Skip to content

Commit 676c616

Browse files
Standardize proposer to z-lab/gemma-4-26B-A4B-it-DFlash: change k3_e2e_gpu_bench default drafter from dflash-kakeya-baseline to z-lab (Gap-B corrected, official) for a consistent proposer across all #107 entry points
Co-authored-by: FluffyAIcode <FluffyAIcode@users.noreply.github.com>
1 parent b909142 commit 676c616

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

scripts/research/k3_e2e_gpu_bench.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
HF_HOME=/workspace/.hf_home PYTHONPATH=.:sdks/python \
2020
.venv-k3/bin/python scripts/research/k3_e2e_gpu_bench.py \
2121
--verifier-id google/gemma-4-26B-A4B-it \
22-
--drafter-id models/dflash-kakeya-baseline \
22+
--drafter-id z-lab/gemma-4-26B-A4B-it-DFlash \
2323
--f-theta-dir results/research/f_theta_v5_s5_sliding \
2424
--haystack-lines 60,160 --n-samples 3 --gen-tokens 16 \
2525
--output results/research/k3_e2e_gpu_bench.json
@@ -167,7 +167,7 @@ def run_restored(adapter, ids_list, samples, gen_tokens, tokenizer, device) -> D
167167
def main() -> int:
168168
ap = argparse.ArgumentParser(description=__doc__)
169169
ap.add_argument("--verifier-id", default="google/gemma-4-26B-A4B-it")
170-
ap.add_argument("--drafter-id", default="models/dflash-kakeya-baseline")
170+
ap.add_argument("--drafter-id", default="z-lab/gemma-4-26B-A4B-it-DFlash")
171171
ap.add_argument("--f-theta-dir", default="results/research/f_theta_v5_s5_sliding")
172172
ap.add_argument("--haystack-lines", default="60,160",
173173
help="Comma-separated haystack line counts (context rungs).")

0 commit comments

Comments
 (0)