Runtimeerror in ubuntu

[liuyuan0546]

Hi, the file is installed by conda but run the test script report error. detail error information as follows:

Loading 4 IgFold models...
Using device: cuda:0
Loading /home/lyw/.local/lib/python3.8/site-packages/igfold/trained_models/IgFold/igfold_1.ckpt...
/home/lyw/.local/lib/python3.8/site-packages/torch/cuda/__init__.py:132: UserWarning: 
    Found GPU0 NVIDIA GeForce GT 720 which is of cuda capability 3.5.
    PyTorch no longer supports this GPU because it is too old.
    The minimum cuda capability supported by this library is 3.7.
    
  warnings.warn(old_gpu_warn % (d, name, major, minor, min_arch // 10, min_arch % 10))
Loading /home/lyw/.local/lib/python3.8/site-packages/igfold/trained_models/IgFold/igfold_2.ckpt...
Loading /home/lyw/.local/lib/python3.8/site-packages/igfold/trained_models/IgFold/igfold_3.ckpt...
Loading /home/lyw/.local/lib/python3.8/site-packages/igfold/trained_models/IgFold/igfold_5.ckpt...
Successfully loaded 4 IgFold models.
Loaded AntiBERTy model.
Traceback (most recent call last):
  File "test.py", line 9, in <module>
    igfold.fold(
  File "/home/lyw/.local/lib/python3.8/site-packages/igfold/IgFoldRunner.py", line 106, in fold
    model_out = fold(
  File "/home/lyw/.local/lib/python3.8/site-packages/igfold/utils/folding.py", line 182, in fold
    embeddings, attentions = antiberty.embed(
  File "/home/lyw/.local/lib/python3.8/site-packages/antiberty/AntiBERTyRunner.py", line 81, in embed
    embeddings[i] = embeddings[i][:, a == 1]
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)