From ed0ca2c960b9b86816b9f0a2a6d6b8722f1217b3 Mon Sep 17 00:00:00 2001
From: haixuanTao <tao.xavier@outlook.com>
Date: Sun, 25 Aug 2024 04:02:26 +0200
Subject: [PATCH] Fix import and add audio ignoring

---
 .gitignore                                  | 2 ++
 node-hub/dora-distil-whisper/pyproject.toml | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/.gitignore b/.gitignore
index 0bf73499..5782040a 100644
--- a/.gitignore
+++ b/.gitignore
@@ -11,6 +11,8 @@ examples/**/*.txt
 
 # Removing images.
 *.jpg
+*.mp3
+*.wav
 *.png
 !docs/src/latency.png
 
diff --git a/node-hub/dora-distil-whisper/pyproject.toml b/node-hub/dora-distil-whisper/pyproject.toml
index e232f3aa..12ceebbc 100644
--- a/node-hub/dora-distil-whisper/pyproject.toml
+++ b/node-hub/dora-distil-whisper/pyproject.toml
@@ -17,6 +17,8 @@ dora-rs = "0.3.5"
 numpy = "< 2.0.0"
 pyarrow = ">= 5.0.0"
 transformers = ">= 4.0.0"
+accelerate = "^0.29.2"
+torch = "^2.1.1"
 
 [tool.poetry.scripts]
 dora-distil-whisper = "dora_distil_whisper.main:main"