runtime error

Exit code: 1. Reason: 16.0G [00:48<00:00, 1.38GB/s] Download complete: 100%|██████████| 16.0G/16.0G [01:35<00:00, 168MB/s] Loading weights: 0%| | 0/291 [00:00<?, ?it/s] Loading weights: 100%|██████████| 291/291 [00:00<00:00, 687.07it/s] Traceback (most recent call last): File "/app/app.py", line 14, in <module> model = AutoModel.from_pretrained('GSAI-ML/LLaDA-8B-Instruct', trust_remote_code=True, ~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ torch_dtype=torch.bfloat16).to(device) ^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.13/site-packages/transformers/models/auto/auto_factory.py", line 380, in from_pretrained return model_class.from_pretrained( ~~~~~~~~~~~~~~~~~~~~~~~~~~~^ pretrained_model_name_or_path, *model_args, config=config, **hub_kwargs, **kwargs ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ ) ^ File "/usr/local/lib/python3.13/site-packages/transformers/modeling_utils.py", line 4133, in from_pretrained loading_info = cls._finalize_model_loading(model, load_config, loading_info) File "/usr/local/lib/python3.13/site-packages/transformers/modeling_utils.py", line 4275, in _finalize_model_loading model.mark_tied_weights_as_initialized(loading_info) ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^ File "/usr/local/lib/python3.13/site-packages/transformers/modeling_utils.py", line 4615, in mark_tied_weights_as_initialized for tied_param in self.all_tied_weights_keys.keys(): ^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.13/site-packages/torch/nn/modules/module.py", line 1968, in __getattr__ raise AttributeError( f"'{type(self).__name__}' object has no attribute '{name}'" ) AttributeError: 'LLaDAModelLM' object has no attribute 'all_tied_weights_keys'. Did you mean: '_tied_weights_keys'?

Container logs: