runtime error

Exit code: 1. Reason: .adaLN_modulation.1.weight', 'encoder.layer.3.adaLN_modulation.1.bias', 'encoder.layer.3.adaLN_modulation.1.weight', 'encoder.layer.4.adaLN_modulation.1.bias', 'encoder.layer.4.adaLN_modulation.1.weight', 'encoder.layer.5.adaLN_modulation.1.bias', 'encoder.layer.5.adaLN_modulation.1.weight', 'encoder.layer.6.adaLN_modulation.1.bias', 'encoder.layer.6.adaLN_modulation.1.weight', 'encoder.layer.7.adaLN_modulation.1.bias', 'encoder.layer.7.adaLN_modulation.1.weight', 'encoder.layer.8.adaLN_modulation.1.bias', 'encoder.layer.8.adaLN_modulation.1.weight', 'encoder.layer.9.adaLN_modulation.1.bias', 'encoder.layer.9.adaLN_modulation.1.weight'] You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference. preprocessor_config.json: 0%| | 0.00/244 [00:00<?, ?B/s] preprocessor_config.json: 100%|██████████| 244/244 [00:00<00:00, 1.31MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 25, in <module> from infer_api import InferAPI File "/home/user/app/infer_api.py", line 694, in <module> infer_slrm_model.init_flexicubes_geometry(infer_slrm_device, fovy=30.0, is_ortho=infer_slrm_model.is_ortho) File "/home/user/app/slrm/models/lrm_mesh.py", line 119, in init_flexicubes_geometry renderer = NeuralRender(device, camera_model=camera) File "/home/user/app/slrm/models/geometry/render/neural_render.py", line 74, in __init__ self.ctx = dr.RasterizeCudaContext(device=device) File "/usr/local/lib/python3.10/site-packages/nvdiffrast/torch/ops.py", line 89, in __init__ self.cpp_wrapper = _get_plugin().RasterizeCRStateWrapper(cuda_device_idx) RuntimeError: CUDA error: no CUDA-capable device is detected CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1. Device-side assertions were explicitly omitted for this error check; the error probably arose while initializing the DSA handlers.

Container logs: