Commit d5058d9
fix: guard device_get with hasattr check for already-converted lists
next_token_ids may already be a Python list (e.g. from overlap path
or spec decoding). Check for .device attribute before calling
jax.device_get().
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>1 parent c5f6f4e commit d5058d9
1 file changed
Lines changed: 4 additions & 2 deletions
Lines changed: 4 additions & 2 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
77 | 77 | | |
78 | 78 | | |
79 | 79 | | |
80 | | - | |
| 80 | + | |
| 81 | + | |
81 | 82 | | |
82 | 83 | | |
83 | 84 | | |
| |||
312 | 313 | | |
313 | 314 | | |
314 | 315 | | |
315 | | - | |
| 316 | + | |
| 317 | + | |
316 | 318 | | |
317 | 319 | | |
318 | 320 | | |
| |||
0 commit comments