chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix: Ensure integer tensor indexing in get_wildcard_emission()

Browse Source
This commit is contained in:
Howard
2025-05-15 14:30:55 +08:00
committed by Barabazs
parent ffedc5cdf0
commit e0833da5dc
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
whisperx/alignment.py
View File

@ -424,7 +424,7 @@ def get_wildcard_emission(frame_emission, tokens, blank_id):
wildcard_mask = (tokens == -1) wildcard_mask = (tokens == -1)
# Get scores for non-wildcard positions # Get scores for non-wildcard positions
regular_scores = frame_emission[tokens.clamp(min=0)] # clamp to avoid -1 index regular_scores = frame_emission[tokens.clamp(min=0).long()] # clamp to avoid -1 index
# Create a mask and compute the maximum value without modifying frame_emission # Create a mask and compute the maximum value without modifying frame_emission
max_valid_score = frame_emission.clone() # Create a copy max_valid_score = frame_emission.clone() # Create a copy