runtime error

Exit code: 1. Reason: ========== == CUDA == ========== CUDA Version 12.1.0 Container image Copyright (c) 2016-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved. This container image and its contents are governed by the NVIDIA Deep Learning Container License. By pulling and using the container, you accept the terms and conditions 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/ . ************************* ** DEPRECATION NOTICE! ** ************************* THIS IMAGE IS DEPRECATED and is scheduled for DELETION. https://gitlab.com/nvidia/container-images/cuda/blob/master/doc/support-policy.md ============================================================ D1337 CIPHER - Training D1337 SOVEREIGN LABS ============================================================ HF_TOKEN: Not found WARNING: No GPU! Loading dataset: pacman1337/d1337-cipher-dataset Generating train split: 0%| | 0/237 [00:00<?, ? examples/s] Generating train split: 100%|██████████| 237/237 [00:00<00:00, 48984.87 examples/s] Dataset: 237 samples Traceback (most recent call last): File "/app/train.py", line 145, in <module> main() File "/app/train.py", line 67, in main sft_config = SFTConfig( File "<string>", line 130, in __init__ File "/home/user/.local/lib/python3.10/site-packages/trl/trainer/sft_config.py", line 289, in __post_init__ super().__post_init__() File "/home/user/.local/lib/python3.10/site-packages/transformers/training_args.py", line 1540, in __post_init__ 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: