runtime error

Exit code: 1. Reason: /usr/local/lib/python3.13/site-packages/transformers/utils/generic.py:441: FutureWarning: `torch.utils._pytree._register_pytree_node` is deprecated. Please use `torch.utils._pytree.register_pytree_node` instead. _torch_pytree._register_pytree_node( Loading medical datasets... train.jsonl: 0%| | 0.00/3.08M [00:00<?, ?B/s] train.jsonl: 100%|██████████| 3.08M/3.08M [00:00<00:00, 65.0MB/s] Generating train split: 0%| | 0/5202 [00:00<?, ? examples/s] Generating train split: 100%|██████████| 5202/5202 [00:00<00:00, 336346.07 examples/s] Generating train split: 0 examples [00:00, ? examples/s] Generating train split: 5202 examples [00:00, 323566.99 examples/s] Starting Fine-tuning (this may take a while on CPU)... Traceback (most recent call last): File "/app/app.py", line 84, in <module> training_args = TrainingArguments( ^^^^^^^^^^^^^^^^^ NameError: name 'TrainingArguments' is not defined

Container logs: