runtime error

Exit code: 1. Reason: nk-cu12, nvidia-nccl-cu12, nvidia-curand-cu12, nvidia-cufft-cu12, nvidia-cuda-runtime-cu12, nvidia-cuda-nvrtc-cu12, nvidia-cuda-cupti-cu12, nvidia-cublas-cu12, networkx, markdown, lightning-utilities, grpcio, absl-py, tensorboard, nvidia-cusparse-cu12, nvidia-cudnn-cu12, huggingface-hub, fire, tokenizers, nvidia-cusolver-cu12, transformers, torch, torchmetrics, pytorch-lightning, aitextgen Attempting uninstall: huggingface-hub Found existing installation: huggingface-hub 0.20.3 Uninstalling huggingface-hub-0.20.3: Successfully uninstalled huggingface-hub-0.20.3 Successfully installed absl-py-2.1.0 aitextgen-0.6.0 fire-0.7.0 grpcio-1.69.0 huggingface-hub-0.27.1 lightning-utilities-0.11.9 markdown-3.7 mpmath-1.3.0 networkx-3.4.2 nvidia-cublas-cu12-12.4.5.8 nvidia-cuda-cupti-cu12-12.4.127 nvidia-cuda-nvrtc-cu12-12.4.127 nvidia-cuda-runtime-cu12-12.4.127 nvidia-cudnn-cu12-9.1.0.70 nvidia-cufft-cu12-11.2.1.3 nvidia-curand-cu12-10.3.5.147 nvidia-cusolver-cu12-11.6.1.9 nvidia-cusparse-cu12-12.3.1.170 nvidia-nccl-cu12-2.21.5 nvidia-nvjitlink-cu12-12.4.127 nvidia-nvtx-cu12-12.4.127 pyDeprecate-0.3.2 pytorch-lightning-1.7.0 regex-2024.11.6 safetensors-0.5.2 sympy-1.13.1 tensorboard-2.18.0 tensorboard-data-server-0.7.2 termcolor-2.5.0 tokenizers-0.21.0 torch-2.5.1 torchmetrics-1.6.1 transformers-4.48.0 triton-3.1.0 werkzeug-3.1.3 [notice] A new release of pip available: 22.3.1 -> 24.3.1 [notice] To update, run: python3 -m pip install --upgrade pip Traceback (most recent call last): File "/home/user/app/app.py", line 31, in <module> from aitextgen import aitextgen File "/home/user/.local/lib/python3.10/site-packages/aitextgen/__init__.py", line 1, in <module> from .aitextgen import aitextgen # noqa File "/home/user/.local/lib/python3.10/site-packages/aitextgen/aitextgen.py", line 25, in <module> from transformers.models.gpt2.convert_gpt2_original_tf_checkpoint_to_pytorch import ( ModuleNotFoundError: No module named 'transformers.models.gpt2.convert_gpt2_original_tf_checkpoint_to_pytorch'

Container logs: