runtime error

Exit code: 1. Reason: WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager, possibly rendering your system unusable. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv. Use the --root-user-action option if you know what you are doing and want to suppress this warning. /usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py:1112: UserWarning: Can't initialize NVML raw_cnt = _raw_device_count_nvml() Failed to load CPU gemm_4bit_forward from kernels-community: No module named 'kernels'. Please make sure you already `pip install kernels` and the kernels >= 0.11.1 /usr/local/lib/python3.10/site-packages/torch/cuda/__init__.py:1112: UserWarning: Can't initialize NVML raw_cnt = _raw_device_count_nvml() Loading tokenizer... Loading base model... Traceback (most recent call last): File "/app/app.py", line 40, in <module> base_model = AutoModelForCausalLM.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 564, in from_pretrained return model_class.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 3452, in from_pretrained hf_quantizer.validate_environment( File "/usr/local/lib/python3.10/site-packages/transformers/quantizers/quantizer_bnb_4bit.py", line 82, in validate_environment validate_bnb_backend_availability(raise_exception=True) File "/usr/local/lib/python3.10/site-packages/transformers/integrations/bitsandbytes.py", line 557, in validate_bnb_backend_availability return _validate_bnb_multi_backend_availability(raise_exception) File "/usr/local/lib/python3.10/site-packages/transformers/integrations/bitsandbytes.py", line 498, in _validate_bnb_multi_backend_availability available_devices.discard("cpu") # Only Intel CPU is supported by BNB at the moment AttributeError: 'frozenset' object has no attribute 'discard'

Container logs: