mchong6 / JoJoGAN

Official PyTorch repo for JoJoGAN: One Shot Face Stylization

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

IndexError: list index out of range

AK391 opened this issue · comments

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

Should be fixed now!