warmup

app.py

@@ -98,20 +98,21 @@ def transcribe(inputs):
         return result
 
     pipe._forward = _forward_time
+    print(inputs)
     text = pipe(inputs, batch_size=BATCH_SIZE)["text"]
 
     yield distil_text, distil_runtime, text, runtime
 
 
-print("Warming up...")
-inputs = np.random.randn(30 * pipe.feature_extractor.sampling_rate) 
-inputs = {"array": inputs, "sampling_rate": pipe.feature_extractor.sampling_rate}
+# print("Warming up...")
+# inputs = np.random.randn(30 * pipe.feature_extractor.sampling_rate) 
+# inputs = {"array": inputs, "sampling_rate": pipe.feature_extractor.sampling_rate}
 
-for _ in range(N_WARMUP):
-    _ = pipe_forward(inputs.copy(), batch_size=BATCH_SIZE)["text"]
-    _ = distil_pipe_forward(inputs.copy(), batch_size=BATCH_SIZE)["text"]
-    print(_)
-print("Models warmed up!")
+# for _ in range(N_WARMUP):
+#     _ = pipe_forward(inputs.copy(), batch_size=BATCH_SIZE)["text"]
+#     _ = distil_pipe_forward(inputs.copy(), batch_size=BATCH_SIZE)["text"]
+#     print(_)
+# print("Models warmed up!")