daniel/h3-blackwell-runtime
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix mrope_freqs_cis call (no device kwarg)

Browse source
This commit is contained in:
Daniel Maddern 2026-08-19 22:59:26 +07:00
parent 9fbd6fe77e
commit 45d3e7bc64
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/h3_blackwell_runtime/qwen3vl_text.py
View file

@ -216,7 +216,7 @@ class Qwen3VL32BTextEncoder(nn.Module):
if position_ids is not None: if position_ids is not None:
from .qwen3vl_vision import mrope_freqs_cis from .qwen3vl_vision import mrope_freqs_cis
freqs_cis = mrope_freqs_cis(position_ids, device=hidden_states.device) freqs_cis = mrope_freqs_cis(position_ids.to(hidden_states.device))
for layer_index, layer in enumerate(self.layers): for layer_index, layer in enumerate(self.layers):
hidden_states = layer(hidden_states, freqs_cis) hidden_states = layer(hidden_states, freqs_cis)
if ( if (