-
Notifications
You must be signed in to change notification settings - Fork 30
Description
Traceback (most recent call last):
File "/root/.pyenv/versions/3.8.16/lib/python3.8/runpy.py", line 194, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/root/.pyenv/versions/3.8.16/lib/python3.8/runpy.py", line 87, in _run_code
exec(code, run_globals)
File "/root/.pyenv/versions/3.8.16/lib/python3.8/site-packages/cog/command/openapi_schema.py", line 26, in
app = create_app(predictor_ref)
File "/root/.pyenv/versions/3.8.16/lib/python3.8/site-packages/cog/server/http.py", line 46, in create_app
predictor = load_predictor_from_ref(predictor_ref)
File "/root/.pyenv/versions/3.8.16/lib/python3.8/site-packages/cog/predictor.py", line 91, in load_predictor_from_ref
spec.loader.exec_module(module)
File "", line 843, in exec_module
File "", line 219, in _call_with_frames_removed
File "predict.py", line 11, in
from models.psp import pSp
File "/src/models/psp.py", line 12, in
from models.encoders import psp_encoders
File "/src/models/encoders/psp_encoders.py", line 8, in
from models.stylegan2.model import EqualLinear
File "/src/models/stylegan2/model.py", line 7, in
from models.stylegan2.op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d
File "/src/models/stylegan2/op/init.py", line 1, in
from .fused_act import FusedLeakyReLU, fused_leaky_relu
File "/src/models/stylegan2/op/fused_act.py", line 9, in
fused = load(
File "/root/.pyenv/versions/3.8.16/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1079, in load
return _jit_compile(
File "/root/.pyenv/versions/3.8.16/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1292, in _jit_compile
_write_ninja_file_and_build_library(
File "/root/.pyenv/versions/3.8.16/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1391, in _write_ninja_file_and_build_library
_write_ninja_file_to_build_library(
File "/root/.pyenv/versions/3.8.16/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1782, in _write_ninja_file_to_build_library
cuda_flags = common_cflags + COMMON_NVCC_FLAGS + _get_cuda_arch_flags()
File "/root/.pyenv/versions/3.8.16/lib/python3.8/site-packages/torch/utils/cpp_extension.py", line 1561, in _get_cuda_arch_flags
arch_list[-1] += '+PTX'
IndexError: list index out of range
ⅹ Failed to get type signature: exit status 1