format

EleutherAI · Apr 16, 2024 · 4b7fd4f · 4b7fd4f
1 parent bfbb896
commit 4b7fd4f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/amt/audio.py b/amt/audio.py
@@ -23,6 +23,7 @@
 FRAMES_PER_SECOND = SAMPLE_RATE // HOP_LENGTH  # 10ms per audio frame
 TOKENS_PER_SECOND = SAMPLE_RATE // N_SAMPLES_PER_TOKEN  # 20ms per audio token
 
+
 def pad_or_trim(array, length: int = N_SAMPLES, *, axis: int = -1):
     """
     Pad or trim the audio array to N_SAMPLES, as expected by the encoder.
@@ -50,6 +51,7 @@ def pad_or_trim(array, length: int = N_SAMPLES, *, axis: int = -1):
 
     return array
 
+
 # Refactor default params are stored in config.json
 class AudioTransform(torch.nn.Module):
     def __init__(