diff --git a/modules/scgpt/scgpt_v0.2.4/requirements.txt b/modules/scgpt/scgpt_v0.2.4/requirements.txt index 0d431951..8287ef88 100644 --- a/modules/scgpt/scgpt_v0.2.4/requirements.txt +++ b/modules/scgpt/scgpt_v0.2.4/requirements.txt @@ -4,7 +4,7 @@ wget==3.2 ipython==8.15.0 cloudpickle==2.2.1 --extra-index-url https://download.pytorch.org/whl/cu118 -torch==2.0.1+cu118 +torch==2.7.1+cu118 torchvision==0.15.2+cu118 https://github.com/Dao-AILab/flash-attention/releases/download/v2.5.8/flash_attn-2.5.8+cu118torch2.0cxx11abiFALSE-cp311-cp311-linux_x86_64.whl scgpt==0.2.4