diff --git a/bert-gen/requirements.txt b/bert-gen/requirements.txt index 7966a28..0543c3d 100644 --- a/bert-gen/requirements.txt +++ b/bert-gen/requirements.txt @@ -51,7 +51,7 @@ SQLAlchemy==1.4.47 tensorboardX==2.6 tokenizers==0.7.0 tomli==2.0.1 -torch==1.5.1 +torch==1.13.1 torchvision==0.6.0a0+35d732a tqdm==4.65.0 transaction==3.1.0