andhakanoon commited on
Commit
533d988
·
verified ·
1 Parent(s): d7b1c9a

card: real step sweep, serve line without ignored hf-overrides

Browse files
Files changed (1) hide show
  1. README.md +13 -10
README.md CHANGED
@@ -42,7 +42,7 @@ oui-1 are committed to the benchmark repo so the number can be rescored offline.
42
 
43
  | Model | Active params | Complete runs |
44
  |---|---|---|
45
- | **oui-1** (this model, 64 denoising steps) | 4B | **132/184 (71.7%)** |
46
  | Gemma 4 31B | 31B | 86/184 (46.7%) |
47
  | Phi-4 14B | 14B | 81/184 (44.0%) |
48
  | Gemma 4 26B-A4B (autoregressive twin) | 4B | 55/184 (29.9%) |
@@ -60,16 +60,17 @@ channel in 167 of 184 outputs and the ones that think complete at the same rate.
60
  On 100 fresh briefs written after training in the same numbered register (never seen by the
61
  model, one generation each, internal scorer): 69/100.
62
 
63
- Speed, 64 denoising steps, single request on one A100 80GB (vLLM 0.24 FP8, entropy-bound sampler,
64
  wall-clock from request to last byte including the ~5k-token prompt), all 46 benchmark briefs,
65
  median per band: lightest screens (~9 statements) 1.0 s, light (~15) 1.5 s, medium (~30) 2.8 s,
66
  dense (~46) 4.6 s, densest (~64) 5.7 s; across the 46, median 2.6 s, p90 5.7 s, about 206
67
  completion tokens per second of wall-clock (250 to 360 tokens per second after the first canvas).
68
  The sampler stops early on its own (15 to 23 denoising steps per 256-token canvas on average), so
69
- a 32-step cap changes little: 133/184 (72.3%) on the public protocol against 132/184 at 64 steps,
70
- and every band within 0.2 s per screen. Four concurrent requests give about 54 screens a minute
71
- at 170 aggregate tokens per second; on short screens the prompt prefill dominates and does not
72
- batch well.
 
73
 
74
  ## What it was trained on
75
 
@@ -84,13 +85,15 @@ teacher-written answer to them. The training set is not released.
84
 
85
  ## How to use
86
 
87
- Serve with vLLM (diffusion sampler settings matter; these are the ones every number above was
88
- measured with):
 
 
 
89
 
90
  ```bash
91
  vllm serve thesysdev/OUI-1 --trust-remote-code --max-model-len 16384 --quantization fp8 \
92
- --served-model-name oui-1 \
93
- --hf-overrides '{"diffusion_sampler":"entropy_bound","diffusion_entropy_bound":0.1,"diffusion_max_denoising_steps":64}' \
94
  --enable-auto-tool-choice --tool-call-parser gemma4
95
  ```
96
 
 
42
 
43
  | Model | Active params | Complete runs |
44
  |---|---|---|
45
+ | **oui-1** (this model, 48 denoising steps) | 4B | **132/184 (71.7%)** |
46
  | Gemma 4 31B | 31B | 86/184 (46.7%) |
47
  | Phi-4 14B | 14B | 81/184 (44.0%) |
48
  | Gemma 4 26B-A4B (autoregressive twin) | 4B | 55/184 (29.9%) |
 
60
  On 100 fresh briefs written after training in the same numbered register (never seen by the
61
  model, one generation each, internal scorer): 69/100.
62
 
63
+ Speed, 48 denoising steps, single request on one A100 80GB (vLLM 0.24 FP8, entropy-bound sampler,
64
  wall-clock from request to last byte including the ~5k-token prompt), all 46 benchmark briefs,
65
  median per band: lightest screens (~9 statements) 1.0 s, light (~15) 1.5 s, medium (~30) 2.8 s,
66
  dense (~46) 4.6 s, densest (~64) 5.7 s; across the 46, median 2.6 s, p90 5.7 s, about 206
67
  completion tokens per second of wall-clock (250 to 360 tokens per second after the first canvas).
68
  The sampler stops early on its own (15 to 23 denoising steps per 256-token canvas on average), so
69
+ the step cap matters little above 32: on the public protocol 32 steps score 134/184, 48 (the
70
+ checkpoint default, used for every number here) 132/184, 64 steps 126/184; 16 steps collapse to
71
+ 59/184 because the forced commit breaks the wiring. Four concurrent requests give about 54 screens
72
+ a minute at 170 aggregate tokens per second; on short screens the prompt prefill dominates and
73
+ does not batch well.
74
 
75
  ## What it was trained on
76
 
 
85
 
86
  ## How to use
87
 
88
+ Serve with vLLM 0.24 or newer. The sampler settings come from the checkpoint's own
89
+ `generation_config.json` (entropy-bound sampler, entropy bound 0.1, 48 denoising steps, 256-token
90
+ canvas); every number above was measured with them. To change the step cap use
91
+ `--diffusion-config '{"canvas_length":256,"max_denoising_steps":N}'`; note that `--hf-overrides`
92
+ keys such as `diffusion_max_denoising_steps` are not read by vLLM.
93
 
94
  ```bash
95
  vllm serve thesysdev/OUI-1 --trust-remote-code --max-model-len 16384 --quantization fp8 \
96
+ --served-model-name oui-1 --max-num-seqs 4 \
 
97
  --enable-auto-tool-choice --tool-call-parser gemma4
98
  ```
99