Skip to content

Conversation

@dayllon-balto
Copy link

In some cases offline diarization crashes with the next error:

RuntimeError: false INTERNAL ASSERT FAILED at "/pytorch/aten/src/ATen/native/BatchLinearAlgebra.cpp":1601, please report a bug to PyTorch. linalg.eigh: Argument 8 has illegal value. Most certainly there is a bug in the implementation calling the backend library.

I noticed this happens with torch 2.6 but doesn't happen with torch 2.3.

This is a fix to avoid this error, making sure that the input matrix for the eigen decomposition has the right shape.

Also adding megatron-core dependency

@github-actions github-actions bot added the ASR label Oct 7, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant