runtime error

Exit code: 1. Reason: 0%| | 0.00/648 [00:00<?, ?B/s] config.json: 100%|██████████| 648/648 [00:00<00:00, 3.18MB/s] tf_model.h5: 0%| | 0.00/537M [00:00<?, ?B/s] tf_model.h5: 10%|█ | 54.3M/537M [00:01<00:09, 52.4MB/s] tf_model.h5: 100%|█████████▉| 537M/537M [00:02<00:00, 260MB/s] 2025-04-08 10:42:07.420190: 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 model checkpoint layers were used when initializing TFBertForMaskedLM. All the layers of TFBertForMaskedLM were initialized from the model checkpoint at dccuchile/bert-base-spanish-wwm-cased. If your task is similar to the task the model of the checkpoint was trained on, you can already use TFBertForMaskedLM for predictions without further training. tokenizer_config.json: 0%| | 0.00/364 [00:00<?, ?B/s] tokenizer_config.json: 100%|██████████| 364/364 [00:00<00:00, 899kB/s] vocab.txt: 0%| | 0.00/242k [00:00<?, ?B/s] vocab.txt: 100%|██████████| 242k/242k [00:00<00:00, 64.6MB/s] tokenizer.json: 0%| | 0.00/480k [00:00<?, ?B/s] tokenizer.json: 100%|██████████| 480k/480k [00:00<00:00, 57.7MB/s] special_tokens_map.json: 0%| | 0.00/134 [00:00<?, ?B/s] special_tokens_map.json: 100%|██████████| 134/134 [00:00<00:00, 1.09MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 4, in <module> fill_mask = pipeline("fill-mask", model="dccuchile/bert-base-spanish-wwm-cased") 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/base.py", line 984, in __init__ if torch.distributed.is_initialized(): NameError: name 'torch' is not defined

Container logs: