runtime error

Exit code: 1. Reason: tions of this license: https://developer.nvidia.com/ngc/nvidia-deep-learning-container-license A copy of this license is made available in this container at /NGC-DL-CONTAINER-LICENSE for your convenience. WARNING: The NVIDIA Driver was not detected. GPU functionality will not be available. Use the NVIDIA Container Toolkit to start this container with GPU support; see https://docs.nvidia.com/datacenter/cloud-native/ . Warning: You are sending unauthenticated requests to the HF Hub. Please set a HF_TOKEN to enable higher rate limits and faster downloads. Loading dataset... Generating train split: 0%| | 0/500057 [00:00<?, ? examples/s] Generating train split: 19%|█▊ | 92900/500057 [00:00<00:02, 184923.07 examples/s] Generating train split: 69%|██████▊ | 342929/500057 [00:01<00:00, 370645.74 examples/s] Generating train split: 87%|████████▋ | 435829/500057 [00:01<00:00, 427802.15 examples/s] Generating train split: 100%|██████████| 500057/500057 [00:01<00:00, 422956.10 examples/s] [transformers] warmup_ratio is deprecated and will be removed in v5.2. Use `warmup_steps` instead. Dataset loaded: 500057 examples Train: 495056, Eval: 5001 Traceback (most recent call last): File "/app/train.py", line 37, in <module> training_args = SFTConfig( File "<string>", line 134, in __init__ File "/usr/local/lib/python3.10/dist-packages/trl/trainer/sft_config.py", line 279, in __post_init__ super().__post_init__() File "/usr/local/lib/python3.10/dist-packages/trl/trainer/base_config.py", line 107, in __post_init__ super().__post_init__() File "/usr/local/lib/python3.10/dist-packages/transformers/training_args.py", line 1576, in __post_init__ self._validate_args() File "/usr/local/lib/python3.10/dist-packages/transformers/training_args.py", line 1738, in _validate_args raise ValueError(error_message) ValueError: Your setup doesn't support bf16/gpu. You need to assign use_cpu if you want to train the model on CPU.

Container logs: