runtime error

Exit code: 1. Reason: 21.0M/1.63G [00:01<01:24, 19.0MB/s] model.safetensors: 5%|▌ | 83.9M/1.63G [00:02<00:36, 42.3MB/s] model.safetensors: 30%|██▉ | 482M/1.63G [00:03<00:05, 198MB/s]  model.safetensors: 75%|███████▍ | 1.22G/1.63G [00:04<00:01, 404MB/s] model.safetensors: 100%|█████████▉| 1.63G/1.63G [00:04<00:00, 339MB/s] 2025-05-17 06:06:17.954640: E external/local_xla/xla/stream_executor/cuda/cuda_platform.cc:51] failed call to cuInit: INTERNAL: CUDA error: Failed call to cuInit: UNKNOWN ERROR (303) All PyTorch model weights were used when initializing TFBartForConditionalGeneration. All the weights of TFBartForConditionalGeneration were initialized from the PyTorch model. If your task is similar to the task the model of the checkpoint was trained on, you can already use TFBartForConditionalGeneration for predictions without further training. vocab.json: 0%| | 0.00/899k [00:00<?, ?B/s] vocab.json: 100%|██████████| 899k/899k [00:00<00:00, 53.6MB/s] merges.txt: 0%| | 0.00/456k [00:00<?, ?B/s] merges.txt: 100%|██████████| 456k/456k [00:00<00:00, 41.7MB/s] tokenizer.json: 0%| | 0.00/1.36M [00:00<?, ?B/s] tokenizer.json: 100%|██████████| 1.36M/1.36M [00:00<00:00, 12.0MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 5, in <module> summarizer = pipeline("summarization", model="facebook/bart-large-cnn") File "/usr/local/lib/python3.10/site-packages/transformers/pipelines/__init__.py", line 1180, in pipeline return pipeline_class(model=model, framework=framework, task=task, **kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/pipelines/text2text_generation.py", line 67, in __init__ super().__init__(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/pipelines/base.py", line 984, in __init__ if torch.distributed.is_initialized(): NameError: name 'torch' is not defined

Container logs: