runtime error

Exit code: 1. Reason: �██▉| 3.25G/3.25G [00:10<00:00, 319MB/s] config.json: 0%| | 0.00/541 [00:00<?, ?B/s] config.json: 100%|██████████| 541/541 [00:00<00:00, 3.12MB/s] model.safetensors: 0%| | 0.00/307M [00:00<?, ?B/s] model.safetensors: 83%|████████▎ | 254M/307M [00:01<00:00, 252MB/s] model.safetensors: 100%|█████████▉| 307M/307M [00:01<00:00, 280MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 20, in <module> MODELS = {name: Zonos.from_pretrained(name, device=device) for name in MODEL_NAMES} File "/home/user/app/app.py", line 20, in <dictcomp> MODELS = {name: Zonos.from_pretrained(name, device=device) for name in MODEL_NAMES} File "/home/user/app/zonos/model.py", line 48, in from_pretrained return cls.from_local(config_path, model_path, device) File "/home/user/app/zonos/model.py", line 53, in from_local model = cls(config).to(device, torch.bfloat16) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1174, in to return self._apply(convert) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 780, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 780, in _apply module._apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 780, 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 805, in _apply param_applied = fn(param) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1160, in convert return t.to( File "/usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py", line 314, 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: