diff --git a/.github/workflows/node_hub_test.sh b/.github/workflows/node_hub_test.sh index 4c14efa2..edc6e832 100755 --- a/.github/workflows/node_hub_test.sh +++ b/.github/workflows/node_hub_test.sh @@ -3,7 +3,7 @@ set -euo # List of ignored modules # TODO: Fix HF github action issue. -ignored_folders=("dora-distil-whisper" "dora-internvl" "dora-parler" "dora-qwenvl" "dora-keyboard" "dora-microphone" "terminal-input") +ignored_folders=("dora-internvl" "dora-parler" "dora-qwenvl" "dora-keyboard" "dora-microphone" "terminal-input") for dir in node-hub/*/ ; do # Get the base name of the directory (without the path) diff --git a/node-hub/dora-distil-whisper/pyproject.toml b/node-hub/dora-distil-whisper/pyproject.toml index 71bf3ecf..4506a2a3 100644 --- a/node-hub/dora-distil-whisper/pyproject.toml +++ b/node-hub/dora-distil-whisper/pyproject.toml @@ -16,7 +16,7 @@ packages = [{ include = "dora_distil_whisper" }] dora-rs = "^0.3.6" numpy = "< 2.0.0" pyarrow = ">= 5.0.0" -transformers = ">= 4.0.0" +transformers = "^4.0.0" accelerate = "^0.29.2" torch = "^2.2.0" python = "^3.7"