diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 84308ae..e350a88 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -2,7 +2,7 @@ aichar tqdm argparse --extra-index-url https://download.pytorch.org/whl/cu118 -torch==2.1.2+cu118 +torch==2.2.2+cu118 torchaudio torchvision torchtext==0.6.0 diff --git a/requirements-webui-cuda.txt b/requirements-webui-cuda.txt index c583b00..7187361 100644 --- a/requirements-webui-cuda.txt +++ b/requirements-webui-cuda.txt @@ -5,7 +5,7 @@ einops wcwidth tqdm --extra-index-url https://download.pytorch.org/whl/cu118 -torch==2.1.2+cu118 +torch==2.2.2+cu118 torchaudio torchvision torchtext==0.6.0 diff --git a/requirements-webui.txt b/requirements-webui.txt index 931ae47..ab329df 100644 --- a/requirements-webui.txt +++ b/requirements-webui.txt @@ -4,7 +4,7 @@ trampoline einops wcwidth tqdm -torch==2.1.2 +torch==2.2.2+cu118 torchvision torchaudio torchtext==0.6.0 diff --git a/requirements.txt b/requirements.txt index 1b1b27c..d463ec7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ aichar tqdm -torch==2.1.2 +torch==2.2.2+cu118 torchvision torchaudio torchtext==0.6.0