From f63154c5505557d25886bd0453b98bb4b0befadb Mon Sep 17 00:00:00 2001 From: Sushmitha Deva Date: Fri, 18 Jul 2025 11:12:55 +0530 Subject: [PATCH 1/2] Pin transformers version Signed-off-by: Sushmitha Deva --- docker/Dockerfile.tts_sdp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile.tts_sdp b/docker/Dockerfile.tts_sdp index ea86ad64..f174c7b1 100644 --- a/docker/Dockerfile.tts_sdp +++ b/docker/Dockerfile.tts_sdp @@ -40,7 +40,7 @@ RUN pip install -r requirements/main.txt RUN pip install -r requirements/tts.txt RUN pip install flash-attn --no-build-isolation RUN pip install https://github.com/LahiLuk/YouTokenToMe/archive/master.zip -RUN pip install megatron-core transformer_engine[pytorch] +RUN pip install megatron-core transformer_engine[pytorch]==2.4.0 RUN pip install nemo_toolkit['all']==2.1.0 WORKDIR /src/NeMo-speech-data-processor \ No newline at end of file From 59f90560408a0e6612cb1285b340f43cbfde315f Mon Sep 17 00:00:00 2001 From: George Zelenfroind Date: Fri, 18 Jul 2025 06:11:38 -0700 Subject: [PATCH 2/2] attempt to fix main Signed-off-by: George Zelenfroind --- requirements/main.txt | 2 ++ 1 file changed, 2 insertions(+) diff --git a/requirements/main.txt b/requirements/main.txt index 12a6c437..99c030b4 100644 --- a/requirements/main.txt +++ b/requirements/main.txt @@ -19,6 +19,8 @@ pydub dask distributed jiwer>=3.1.0,<4.0.0 +pyarrow>=8.0.0,<14.0.0 +datasets>=2.14.0,<3.0.0 # toloka-kit # Temporarily disabled due to Toloka's technical pause; keep as reference for past and future API support # for some processers, additionally https://github.com/NVIDIA/NeMo is required # for some processers, additionally nemo_text_processing is required