diff --git a/requirements.txt b/requirements.txt index b9025b1f..bc3efb22 100644 --- a/requirements.txt +++ b/requirements.txt @@ -12,7 +12,7 @@ tensorboard==2.15.2 tensorflow-io-gcs-filesystem==0.31.0 # PyTorch ecosystem -torch==2.1.0 +torch>=2.1.0 torchaudio==2.1.0 pytorch-lightning==2.1.3 torchmetrics==1.2.0 @@ -30,7 +30,7 @@ onnxruntime==1.16.3 # onnxruntime-gpu==1.16.3 # Uncomment for GPU, comment out onnxruntime above # NLP and transformers -transformers==4.35.2 +transformers>=4.35.0 sentencepiece==0.1.99 peft==0.6.2