Sofia Casadei commited on
Commit
edfee48
Β·
1 Parent(s): 0fa1945

fix use flash attn

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -91,8 +91,8 @@ async def transcribe(audio: tuple[int, np.ndarray]):
91
  logger.info(f"Sample rate: {sample_rate}Hz, Shape: {audio_array.shape}")
92
 
93
  outputs = transcribe_pipeline(
94
- #audio_to_bytes(audio), # pass bytes
95
- audio_array, # pass numpy array
96
  chunk_length_s=3,
97
  batch_size=1,
98
  generate_kwargs={
 
91
  logger.info(f"Sample rate: {sample_rate}Hz, Shape: {audio_array.shape}")
92
 
93
  outputs = transcribe_pipeline(
94
+ audio_to_bytes(audio), # pass bytes
95
+ #audio_array, # pass numpy array
96
  chunk_length_s=3,
97
  batch_size=1,
98
  generate_kwargs={