mchong6 / JoJoGAN

Official PyTorch repo for JoJoGAN: One Shot Face Stylization
MIT License
1.42k stars 206 forks source link

IndexError: list index out of range #2

Closed AK391 closed 2 years ago

AK391 commented 2 years ago

when device is set to cpu in colab and hardware accelerator is none

No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda'

IndexError Traceback (most recent call last)

in () 24 from tqdm import tqdm 25 import lpips ---> 26 from model import * 27 from e4e_projection import projection as e4e_projection 28 from restyle_projection import projection as restyle_projection 7 frames /usr/local/lib/python3.7/dist-packages/torch/utils/cpp_extension.py in _get_cuda_arch_flags(cflags) 1604 arch_list.append(arch) 1605 arch_list = sorted(arch_list) -> 1606 arch_list[-1] += '+PTX' 1607 else: 1608 # Deal with lists that are ' ' separated (only deal with ';' after) IndexError: list index out of range
mchong6 commented 2 years ago

Should be fixed now!

AK391 commented 2 years ago

@mchong6 thanks!