diff --git a/requirements.txt b/requirements.txt index 74aef71..58b52ff 100644 --- a/requirements.txt +++ b/requirements.txt @@ -23,6 +23,6 @@ torchinfo==1.8.0 tqdm==4.67.1 torchaudio==2.7.0 torchvision==0.22.0 -transformers==5.0.0rc3 +transformers==5.3.0 triton==3.3.0 webdataset==0.2.111 \ No newline at end of file diff --git a/wmar_audio/requirements.txt b/wmar_audio/requirements.txt index 33fe313..3cf395e 100644 --- a/wmar_audio/requirements.txt +++ b/wmar_audio/requirements.txt @@ -12,7 +12,7 @@ torch==2.8.0 torchmetrics[audio]==1.7.1 accelerate rouge_score -transformers>=4.40.0 +transformers>=5.3.0 torchaudio tqdm bitsandbytes