runtime error

Exit code: 1. Reason: ██| 335M/335M [00:01<00:00, 185MB/s] scheduler_config.json: 0%| | 0.00/479 [00:00<?, ?B/s] scheduler_config.json: 100%|██████████| 479/479 [00:00<00:00, 5.16MB/s] The config attributes {'clip_sample': False, 'sample_max_value': 1.0, 'set_alpha_to_one': False, 'skip_prk_steps': True} were passed to FlowMatchEulerDiscreteScheduler, but are not expected and will be ignored. Please verify your scheduler_config.json configuration file. Traceback (most recent call last): File "/app/app.py", line 17, in <module> model, birefnet = load_lbm_and_segmenter() ^^^^^^^^^^^^^^^^^^^^^^^^ File "/app/relight_engine.py", line 81, in load_lbm_and_segmenter model.to("cuda").to(torch.bfloat16) ^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.12/site-packages/lbm/models/base/base_model.py", line 40, in to self = super().to( ^^^^^^^^^^^ File "/usr/local/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1355, in to return self._apply(convert) ^^^^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.12/site-packages/torch/nn/modules/module.py", line 915, in _apply module._apply(fn) File "/usr/local/lib/python3.12/site-packages/torch/nn/modules/module.py", line 915, in _apply module._apply(fn) File "/usr/local/lib/python3.12/site-packages/torch/nn/modules/module.py", line 915, in _apply module._apply(fn) [Previous line repeated 1 more time] File "/usr/local/lib/python3.12/site-packages/torch/nn/modules/module.py", line 942, in _apply param_applied = fn(param) ^^^^^^^^^ File "/usr/local/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1341, in convert return t.to( ^^^^^ File "/usr/local/lib/python3.12/site-packages/torch/cuda/__init__.py", line 372, 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: