Andrewsab commited on
Commit
ea78c27
·
verified ·
1 Parent(s): b8078e5

V15 R-3 Win-CUDA cuda-int8-bnb ship-form (Win port of macOS WP#919)

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,136 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language: ru
4
+ library_name: transformers
5
+ base_model: Qwen/Qwen3.5-0.8B
6
+ tags:
7
+ - dictation
8
+ - russian
9
+ - lora
10
+ - VoiceScribe
11
+ - corrector
12
+ - qwen3.5
13
+ datasets:
14
+ - VoiceScribe/dictation-corrector-seed
15
+ metrics:
16
+ - exact-match
17
+ ---
18
+
19
+ # Voice Scribe Russian Dictation Corrector (Qwen3.5-0.8B, V15 R-3, cuda-int8-bnb)
20
+
21
+ Premium ship-form: bitsandbytes 8-bit (LLM.int8) quantization. ~981 MB. ZERO observed quality loss vs bf16 (96.55% identical). Target: RTX 30xx+/8GB+.
22
+
23
+ ## Eval results (held-out wild_eval, 58 prompts × 9 sectors)
24
+
25
+ | Metric | Score |
26
+ |---|---:|
27
+ | Wild pass | 96.55% |
28
+ | Hard-negative | 5/5 |
29
+ | Smoke | 7/8 |
30
+ | p50 latency | 1738 ms |
31
+ | Ship-form size | 981 MB |
32
+
33
+ **Comparison:**
34
+ - macOS V15 R-3 reference: 93.1% wild
35
+ - V14 baseline: 86.2%
36
+ - Qwen3-4B Q5 production (pre-LoRA): 48%
37
+ - This model: **96.55%** (+10.3pp vs V14 baseline)
38
+
39
+ ## Training recipe (V15 R-3)
40
+
41
+ Mirrors macOS `configs/r4_v15_extended.yaml` byte-for-byte logical-recipe.
42
+
43
+ ```
44
+ base = Qwen/Qwen3.5-0.8B (vanilla, NOT Instruct)
45
+ LoRA rank = 16
46
+ LoRA alpha = 80 (rsLoRA mode -> effective scale 20.0)
47
+ target_modules = q_proj, k_proj, v_proj, o_proj
48
+ layers_to_transform = last 16 of 24 (range(8, 24))
49
+ mask_prompt = ON (assistant_masks via patched chat_template {% generation %})
50
+ max_steps = 1100
51
+ batch_size = 2
52
+ max_seq_length = 384
53
+ lr_schedule = cosine, peak 3e-5, warmup 100
54
+ weight_decay = 0.01
55
+ optim = adamw_torch_fused
56
+ precision = bf16
57
+ seed = 20260515
58
+ trainable params = 720,896 (0.0957% of 753M)
59
+ data = 1104 rows = V14 seeds (691) + V15 brand expansion (271) + V15 R-3 patches (142)
60
+ ```
61
+
62
+ ## Intended use
63
+
64
+ - Russian dictation cleanup after ASR (GigaAM, Whisper, Parakeet)
65
+ - Conservative editing policy: remove filler (эм/ну/типа/короче), normalize Cyrillic IT terms (гитхаб -> GitHub), preserve all meaning
66
+ - **NOT** for general text editing, English text, creative writing, summarization, translation
67
+
68
+ ## Limitations
69
+
70
+ - 58-row eval set has ±1.72pp single-row noise
71
+ - Cyrillic <-> Latin choice on ambiguous brand spellings is judgment call (model may differ from expected byte-match)
72
+ - Trained on synthetic data only; real production telemetry collection planned for V16
73
+ - INT8 latency on RTX 50xx Blackwell is sub-optimal (bnb LLM.int8 kernels)
74
+ -
75
+
76
+ ## Hardware ship matrix
77
+
78
+ | Hardware | Recommended ship-form | This model? |
79
+ |---|---|---:|
80
+ | RTX 5090 / 4090 24GB+ | bf16 | |
81
+ | RTX 4070 / 4060 / 3060 8-16GB | INT8 | PRIMARY |
82
+ | RTX 2060 / 3050 / 4060 6-8GB | INT4 NF4 | |
83
+ | Re-training / stacking | adapter | |
84
+
85
+ ## Inference
86
+
87
+ ```python
88
+ from transformers import AutoModelForCausalLM, AutoTokenizer
89
+ import torch
90
+
91
+ model = AutoModelForCausalLM.from_pretrained(
92
+ "VoiceScribe/qwen3-5-0.8b-dictation-corrector-cuda-int8",
93
+ load_in_8bit=True,
94
+ device_map="cuda",
95
+ trust_remote_code=True,
96
+ )
97
+ tokenizer = AutoTokenizer.from_pretrained("VoiceScribe/qwen3-5-0.8b-dictation-corrector-cuda-int8", trust_remote_code=True)
98
+
99
+ messages = [
100
+ {"role": "system", "content": "Корректор русской диктовки. Убери слова-паразиты ..."},
101
+ {"role": "user", "content": "Запушил коммит в гитхаб репозиторий"},
102
+ ]
103
+ prompt = tokenizer.apply_chat_template(
104
+ messages,
105
+ add_generation_prompt=True,
106
+ tokenize=False,
107
+ enable_thinking=False, # CRITICAL for Qwen3.5
108
+ )
109
+ inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
110
+ out = model.generate(**inputs, max_new_tokens=200, do_sample=False)
111
+ print(tokenizer.decode(out[0, inputs["input_ids"].shape[1]:], skip_special_tokens=True))
112
+ # Expected: "Запушил коммит в GitHub репозиторий"
113
+ ```
114
+
115
+ ## Cross-platform variants
116
+
117
+ - **macOS MLX**: `VoiceScribe/qwen3-5-0.8b-dictation-corrector-mlx-{bf16,8bit,4bit}` (V15 R-3, 93.1% wild)
118
+ - **CUDA bf16/INT8/INT4-NF4**: this family (V15 R-3 Win port, 84.48-96.55% wild)
119
+ - **OpenVINO**: planned (separate venv for export; tracker WP#920)
120
+ - **TensorRT-RTX W4A16**: deferred (DeltaNet ONNX export blocked on Win-Py3.13-cu128 in 2026-05)
121
+
122
+ ## Citation
123
+
124
+ ```bibtex
125
+ @software{voicescribe-corrector-v15r3-2026,
126
+ title = {Voice Scribe Russian Dictation Corrector (Qwen3.5-0.8B V15 R-3, CUDA Win port)},
127
+ author = {Sabynin, Andrey},
128
+ year = {2026},
129
+ url = {https://huggingface.co/VoiceScribe/qwen3-5-0.8b-dictation-corrector-cuda-int8}
130
+ }
131
+ ```
132
+
133
+ ## Trackers
134
+
135
+ - macOS R&D: OpenProject WP#917 (V14), WP#919 (V15 R-3 macOS)
136
+ - Windows port: OpenProject WP#920 (this effort, achieved 96.55% vs macOS 93.1%)
chat_template.jinja ADDED
@@ -0,0 +1,154 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- set image_count = namespace(value=0) %}
2
+ {%- set video_count = namespace(value=0) %}
3
+ {%- macro render_content(content, do_vision_count, is_system_content=false) %}
4
+ {%- if content is string %}
5
+ {{- content }}
6
+ {%- elif content is iterable and content is not mapping %}
7
+ {%- for item in content %}
8
+ {%- if 'image' in item or 'image_url' in item or item.type == 'image' %}
9
+ {%- if is_system_content %}
10
+ {{- raise_exception('System message cannot contain images.') }}
11
+ {%- endif %}
12
+ {%- if do_vision_count %}
13
+ {%- set image_count.value = image_count.value + 1 %}
14
+ {%- endif %}
15
+ {%- if add_vision_id %}
16
+ {{- 'Picture ' ~ image_count.value ~ ': ' }}
17
+ {%- endif %}
18
+ {{- '<|vision_start|><|image_pad|><|vision_end|>' }}
19
+ {%- elif 'video' in item or item.type == 'video' %}
20
+ {%- if is_system_content %}
21
+ {{- raise_exception('System message cannot contain videos.') }}
22
+ {%- endif %}
23
+ {%- if do_vision_count %}
24
+ {%- set video_count.value = video_count.value + 1 %}
25
+ {%- endif %}
26
+ {%- if add_vision_id %}
27
+ {{- 'Video ' ~ video_count.value ~ ': ' }}
28
+ {%- endif %}
29
+ {{- '<|vision_start|><|video_pad|><|vision_end|>' }}
30
+ {%- elif 'text' in item %}
31
+ {{- item.text }}
32
+ {%- else %}
33
+ {{- raise_exception('Unexpected item type in content.') }}
34
+ {%- endif %}
35
+ {%- endfor %}
36
+ {%- elif content is none or content is undefined %}
37
+ {{- '' }}
38
+ {%- else %}
39
+ {{- raise_exception('Unexpected content type.') }}
40
+ {%- endif %}
41
+ {%- endmacro %}
42
+ {%- if not messages %}
43
+ {{- raise_exception('No messages provided.') }}
44
+ {%- endif %}
45
+ {%- if tools and tools is iterable and tools is not mapping %}
46
+ {{- '<|im_start|>system\n' }}
47
+ {{- "# Tools\n\nYou have access to the following functions:\n\n<tools>" }}
48
+ {%- for tool in tools %}
49
+ {{- "\n" }}
50
+ {{- tool | tojson }}
51
+ {%- endfor %}
52
+ {{- "\n</tools>" }}
53
+ {{- '\n\nIf you choose to call a function ONLY reply in the following format with NO suffix:\n\n<tool_call>\n<function=example_function_name>\n<parameter=example_parameter_1>\nvalue_1\n</parameter>\n<parameter=example_parameter_2>\nThis is the value for the second parameter\nthat can span\nmultiple lines\n</parameter>\n</function>\n</tool_call>\n\n<IMPORTANT>\nReminder:\n- Function calls MUST follow the specified format: an inner <function=...></function> block must be nested within <tool_call></tool_call> XML tags\n- Required parameters MUST be specified\n- You may provide optional reasoning for your function call in natural language BEFORE the function call, but NOT after\n- If there is no function call available, answer the question like normal with your current knowledge and do not tell the user about function calls\n</IMPORTANT>' }}
54
+ {%- if messages[0].role == 'system' %}
55
+ {%- set content = render_content(messages[0].content, false, true)|trim %}
56
+ {%- if content %}
57
+ {{- '\n\n' + content }}
58
+ {%- endif %}
59
+ {%- endif %}
60
+ {{- '<|im_end|>\n' }}
61
+ {%- else %}
62
+ {%- if messages[0].role == 'system' %}
63
+ {%- set content = render_content(messages[0].content, false, true)|trim %}
64
+ {{- '<|im_start|>system\n' + content + '<|im_end|>\n' }}
65
+ {%- endif %}
66
+ {%- endif %}
67
+ {%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}
68
+ {%- for message in messages[::-1] %}
69
+ {%- set index = (messages|length - 1) - loop.index0 %}
70
+ {%- if ns.multi_step_tool and message.role == "user" %}
71
+ {%- set content = render_content(message.content, false)|trim %}
72
+ {%- if not(content.startswith('<tool_response>') and content.endswith('</tool_response>')) %}
73
+ {%- set ns.multi_step_tool = false %}
74
+ {%- set ns.last_query_index = index %}
75
+ {%- endif %}
76
+ {%- endif %}
77
+ {%- endfor %}
78
+ {%- if ns.multi_step_tool %}
79
+ {{- raise_exception('No user query found in messages.') }}
80
+ {%- endif %}
81
+ {%- for message in messages %}
82
+ {%- set content = render_content(message.content, true)|trim %}
83
+ {%- if message.role == "system" %}
84
+ {%- if not loop.first %}
85
+ {{- raise_exception('System message must be at the beginning.') }}
86
+ {%- endif %}
87
+ {%- elif message.role == "user" %}
88
+ {{- '<|im_start|>' + message.role + '\n' + content + '<|im_end|>' + '\n' }}
89
+ {%- elif message.role == "assistant" %}
90
+ {%- set reasoning_content = '' %}
91
+ {%- if message.reasoning_content is string %}
92
+ {%- set reasoning_content = message.reasoning_content %}
93
+ {%- else %}
94
+ {%- if '</think>' in content %}
95
+ {%- set reasoning_content = content.split('</think>')[0].rstrip('\n').split('<think>')[-1].lstrip('\n') %}
96
+ {%- set content = content.split('</think>')[-1].lstrip('\n') %}
97
+ {%- endif %}
98
+ {%- endif %}
99
+ {%- set reasoning_content = reasoning_content|trim %}
100
+ {%- if loop.index0 > ns.last_query_index %}
101
+ {{- '<|im_start|>' + message.role + '\n<think>\n' + reasoning_content + '\n</think>\n\n' + content }}
102
+ {%- else %}
103
+ {{- '<|im_start|>' + message.role + '\n' + content }}
104
+ {%- endif %}
105
+ {%- if message.tool_calls and message.tool_calls is iterable and message.tool_calls is not mapping %}
106
+ {%- for tool_call in message.tool_calls %}
107
+ {%- if tool_call.function is defined %}
108
+ {%- set tool_call = tool_call.function %}
109
+ {%- endif %}
110
+ {%- if loop.first %}
111
+ {%- if content|trim %}
112
+ {{- '\n\n<tool_call>\n<function=' + tool_call.name + '>\n' }}
113
+ {%- else %}
114
+ {{- '<tool_call>\n<function=' + tool_call.name + '>\n' }}
115
+ {%- endif %}
116
+ {%- else %}
117
+ {{- '\n<tool_call>\n<function=' + tool_call.name + '>\n' }}
118
+ {%- endif %}
119
+ {%- if tool_call.arguments is defined %}
120
+ {%- for args_name, args_value in tool_call.arguments|items %}
121
+ {{- '<parameter=' + args_name + '>\n' }}
122
+ {%- set args_value = args_value | tojson | safe if args_value is mapping or (args_value is sequence and args_value is not string) else args_value | string %}
123
+ {{- args_value }}
124
+ {{- '\n</parameter>\n' }}
125
+ {%- endfor %}
126
+ {%- endif %}
127
+ {{- '</function>\n</tool_call>' }}
128
+ {%- endfor %}
129
+ {%- endif %}
130
+ {{- '<|im_end|>\n' }}
131
+ {%- elif message.role == "tool" %}
132
+ {%- if loop.previtem and loop.previtem.role != "tool" %}
133
+ {{- '<|im_start|>user' }}
134
+ {%- endif %}
135
+ {{- '\n<tool_response>\n' }}
136
+ {{- content }}
137
+ {{- '\n</tool_response>' }}
138
+ {%- if not loop.last and loop.nextitem.role != "tool" %}
139
+ {{- '<|im_end|>\n' }}
140
+ {%- elif loop.last %}
141
+ {{- '<|im_end|>\n' }}
142
+ {%- endif %}
143
+ {%- else %}
144
+ {{- raise_exception('Unexpected message role.') }}
145
+ {%- endif %}
146
+ {%- endfor %}
147
+ {%- if add_generation_prompt %}
148
+ {{- '<|im_start|>assistant\n' }}
149
+ {%- if enable_thinking is defined and enable_thinking is true %}
150
+ {{- '<think>\n' }}
151
+ {%- else %}
152
+ {{- '<think>\n\n</think>\n\n' }}
153
+ {%- endif %}
154
+ {%- endif %}
config.json ADDED
@@ -0,0 +1,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Qwen3_5ForCausalLM"
4
+ ],
5
+ "attention_bias": false,
6
+ "attention_dropout": 0.0,
7
+ "attn_output_gate": true,
8
+ "bos_token_id": null,
9
+ "dtype": "bfloat16",
10
+ "eos_token_id": 248044,
11
+ "full_attention_interval": 4,
12
+ "head_dim": 256,
13
+ "hidden_act": "silu",
14
+ "hidden_size": 1024,
15
+ "initializer_range": 0.02,
16
+ "intermediate_size": 3584,
17
+ "layer_types": [
18
+ "linear_attention",
19
+ "linear_attention",
20
+ "linear_attention",
21
+ "full_attention",
22
+ "linear_attention",
23
+ "linear_attention",
24
+ "linear_attention",
25
+ "full_attention",
26
+ "linear_attention",
27
+ "linear_attention",
28
+ "linear_attention",
29
+ "full_attention",
30
+ "linear_attention",
31
+ "linear_attention",
32
+ "linear_attention",
33
+ "full_attention",
34
+ "linear_attention",
35
+ "linear_attention",
36
+ "linear_attention",
37
+ "full_attention",
38
+ "linear_attention",
39
+ "linear_attention",
40
+ "linear_attention",
41
+ "full_attention"
42
+ ],
43
+ "linear_conv_kernel_dim": 4,
44
+ "linear_key_head_dim": 128,
45
+ "linear_num_key_heads": 16,
46
+ "linear_num_value_heads": 16,
47
+ "linear_value_head_dim": 128,
48
+ "mamba_ssm_dtype": "float32",
49
+ "max_position_embeddings": 262144,
50
+ "mlp_only_layers": [],
51
+ "model_type": "qwen3_5_text",
52
+ "mtp_num_hidden_layers": 1,
53
+ "mtp_use_dedicated_embeddings": false,
54
+ "num_attention_heads": 8,
55
+ "num_hidden_layers": 24,
56
+ "num_key_value_heads": 2,
57
+ "pad_token_id": null,
58
+ "partial_rotary_factor": 0.25,
59
+ "quantization_config": {
60
+ "_load_in_4bit": false,
61
+ "_load_in_8bit": true,
62
+ "bnb_4bit_compute_dtype": "float32",
63
+ "bnb_4bit_quant_storage": "uint8",
64
+ "bnb_4bit_quant_type": "fp4",
65
+ "bnb_4bit_use_double_quant": false,
66
+ "llm_int8_enable_fp32_cpu_offload": false,
67
+ "llm_int8_has_fp16_weight": false,
68
+ "llm_int8_skip_modules": null,
69
+ "llm_int8_threshold": 6.0,
70
+ "load_in_4bit": false,
71
+ "load_in_8bit": true,
72
+ "quant_method": "bitsandbytes"
73
+ },
74
+ "rms_norm_eps": 1e-06,
75
+ "rope_parameters": {
76
+ "mrope_interleaved": true,
77
+ "mrope_section": [
78
+ 11,
79
+ 11,
80
+ 10
81
+ ],
82
+ "partial_rotary_factor": 0.25,
83
+ "rope_theta": 10000000,
84
+ "rope_type": "default"
85
+ },
86
+ "tie_word_embeddings": true,
87
+ "transformers_version": "5.8.1",
88
+ "use_cache": true,
89
+ "vocab_size": 248320
90
+ }
generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "eos_token_id": 248044,
4
+ "transformers_version": "5.8.1",
5
+ "use_cache": true
6
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:13942aff55afb34d997e10ad0e73de854cb0589e47a8d2e40188c3920e2d0de6
3
+ size 1008857842
quant_info.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "base": "Qwen/Qwen3.5-0.8B",
3
+ "adapter": "rnd\\wp917-win-lora\\outputs\\v15-r3-seed20260515",
4
+ "bits": 8,
5
+ "size_mb": 981.2
6
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:06b9509352d2af50381ab2247e083b80d32d5c0aba91c272ca9ff729b6a0e523
3
+ size 19989325
tokenizer_config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "audio_bos_token": "<|audio_start|>",
4
+ "audio_eos_token": "<|audio_end|>",
5
+ "audio_token": "<|audio_pad|>",
6
+ "backend": "tokenizers",
7
+ "bos_token": null,
8
+ "clean_up_tokenization_spaces": false,
9
+ "eos_token": "<|im_end|>",
10
+ "errors": "replace",
11
+ "image_token": "<|image_pad|>",
12
+ "is_local": false,
13
+ "local_files_only": false,
14
+ "model_max_length": 262144,
15
+ "model_specific_special_tokens": {
16
+ "audio_bos_token": "<|audio_start|>",
17
+ "audio_eos_token": "<|audio_end|>",
18
+ "audio_token": "<|audio_pad|>",
19
+ "image_token": "<|image_pad|>",
20
+ "video_token": "<|video_pad|>",
21
+ "vision_bos_token": "<|vision_start|>",
22
+ "vision_eos_token": "<|vision_end|>"
23
+ },
24
+ "pad_token": "<|endoftext|>",
25
+ "pretokenize_regex": "(?i:'s|'t|'re|'ve|'m|'ll|'d)|[^\\r\\n\\p{L}\\p{N}]?[\\p{L}\\p{M}]+|\\p{N}| ?[^\\s\\p{L}\\p{M}\\p{N}]+[\\r\\n]*|\\s*[\\r\\n]+|\\s+(?!\\S)|\\s+",
26
+ "split_special_tokens": false,
27
+ "tokenizer_class": "Qwen2Tokenizer",
28
+ "unk_token": null,
29
+ "video_token": "<|video_pad|>",
30
+ "vision_bos_token": "<|vision_start|>",
31
+ "vision_eos_token": "<|vision_end|>"
32
+ }