runtime error

Exit code: 1. Reason: ▌ | 1.61G/4.55G [00:08<00:12, 230MB/s] model.safetensors: 49%|████▉ | 2.22G/4.55G [00:10<00:07, 315MB/s] model.safetensors: 59%|█████▉ | 2.68G/4.55G [00:11<00:05, 335MB/s] model.safetensors: 68%|██████▊ | 3.09G/4.55G [00:12<00:04, 329MB/s] model.safetensors: 82%|████████▏ | 3.74G/4.55G [00:13<00:01, 407MB/s] model.safetensors: 97%|█████████▋| 4.41G/4.55G [00:14<00:00, 470MB/s] model.safetensors: 100%|██████████| 4.55G/4.55G [00:14<00:00, 305MB/s] Number of parameters in VAE: 340.30M Loaded VAE from /home/user/.cache/huggingface/hub/models--nvidia--PartPacker/snapshots/17046fe1729d05db99cb11d7cc6885d777fcfc73/vae.pt Number of parameters in DiT: 1249.51M Traceback (most recent call last): File "/home/user/app/app.py", line 57, in <module> model = Model(model_config).eval().cuda().bfloat16() File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1050, in cuda return self._apply(lambda t: t.cuda(device)) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 900, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 900, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 900, in _apply module._apply(fn) [Previous line repeated 1 more time] File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 927, in _apply param_applied = fn(param) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1050, in <lambda> return self._apply(lambda t: t.cuda(device)) File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 319, in _lazy_init torch._C._cuda_init() RuntimeError: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx

Container logs: