transformers
5e4d6304 - [NemotronH-Omni] Fix device mismatch in test tensor creation (#49102)

Commit
14 days ago
[NemotronH-Omni] Fix device mismatch in test tensor creation (#49102) * [NemotronH-Omni] Fix device mismatch in test tensor creation `image_grid_hw`, `positions`, and `input_features_mask` were created without `device=torch_device`, causing RuntimeError when tests run on GPU: CPU tensors collided with GPU model buffers (e.g. `summary_idxs` in RadioModel._forward_packed, `lengths` in create_audio_mask). Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> * style: ruff format test_modeling_nemotron_h_omni.py * Skip ONNX/torch export tests for NemotronHOmniVision2TextModelTest (data-dependent shapes) --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
Author
Parents
Loading