Skip to content

Commit

Permalink
Fix dtype for context array
Browse files Browse the repository at this point in the history
  • Loading branch information
hoonlight committed Jun 28, 2024
1 parent 91c3890 commit ed87016
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion faster_whisper/vad.py
Original file line number Diff line number Diff line change
Expand Up @@ -251,7 +251,7 @@ def __init__(self, path):

def get_initial_state(self, batch_size: int):
state = np.zeros((2, batch_size, 128), dtype=np.float32)
context = np.zeros((batch_size, 64))
context = np.zeros((batch_size, 64), dtype=np.float32)
return state, context

def __call__(self, x, state, context, sr: int):
Expand Down

0 comments on commit ed87016

Please sign in to comment.