Commit 337757c
authored
[BUGFIX] Patch for issues with export via replicate_kv_heads script CLI for CB (#646)
InputHandler has changes to create position_ids based on CB batch size.
Signed-off-by: Dhiraj Kumar Sah <[email protected]>1 parent 529b530 commit 337757c
File tree
2 files changed
+3
-2
lines changed- QEfficient
- exporter
- utils
2 files changed
+3
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
202 | 202 | | |
203 | 203 | | |
204 | 204 | | |
205 | | - | |
| 205 | + | |
206 | 206 | | |
207 | 207 | | |
208 | 208 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
68 | 68 | | |
69 | 69 | | |
70 | 70 | | |
71 | | - | |
| 71 | + | |
| 72 | + | |
72 | 73 | | |
73 | 74 | | |
74 | 75 | | |
| |||
0 commit comments