runtime error

Exit code: 1. Reason: ?B/s] config.json: 100%|██████████| 846/846 [00:00<00:00, 6.94MB/s] pytorch_model.bin: 0%| | 0.00/2.46G [00:00<?, ?B/s] pytorch_model.bin: 3%|▎ | 73.4M/2.46G [00:01<00:37, 64.1MB/s] pytorch_model.bin: 18%|█▊ | 440M/2.46G [00:02<00:08, 231MB/s]  pytorch_model.bin: 43%|████▎ | 1.06G/2.46G [00:03<00:03, 401MB/s] pytorch_model.bin: 67%|██████▋ | 1.65G/2.46G [00:04<00:01, 468MB/s] pytorch_model.bin: 100%|█████████▉| 2.46G/2.46G [00:05<00:00, 467MB/s] Traceback (most recent call last): File "/home/user/app/run.py", line 6, in <module> model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M") File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 573, in from_pretrained return model_class.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 269, in _wrapper return func(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 4366, in from_pretrained model_init_context = cls.get_init_context(is_quantized, _is_ds_init_called) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 3769, in get_init_context init_contexts = [no_init_weights(), init_empty_weights()] NameError: name 'init_empty_weights' is not defined model.safetensors: 0%| | 0.00/2.46G [00:00<?, ?B/s] model.safetensors: 1%|▏ | 31.5M/2.46G [00:01<01:18, 31.1MB/s] model.safetensors: 4%|▍ | 94.4M/2.46G [00:02<00:49, 47.7MB/s] model.safetensors: 41%|████ | 1.01G/2.46G [00:03<00:03, 423MB/s]  model.safetensors: 58%|█████▊ | 1.44G/2.46G [00:04<00:02, 377MB/s] model.safetensors: 91%|█████████ | 2.24G/2.46G [00:05<00:00, 509MB/s] model.safetensors: 100%|█████████▉| 2.46G/2.46G [00:05<00:00, 417MB/s]

Container logs: