runtime error

Exit code: 1. Reason: ers/modeling_utils.py", line 2534, in smart_apply smart_apply(child, child._initialize_weights, is_remote_code) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 2536, in smart_apply smart_apply(child, fn, is_remote_code) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 2536, in smart_apply smart_apply(child, fn, is_remote_code) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 2536, in smart_apply smart_apply(child, fn, is_remote_code) [Previous line repeated 3 more times] File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 2537, in smart_apply fn(module, is_remote_code) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 2514, in _initialize_weights self._init_weights(module) File "/usr/local/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 124, in decorate_context return func(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/models/llama4/modeling_llama4.py", line 478, in _init_weights super()._init_weights(module) File "/usr/local/lib/python3.10/site-packages/torch/utils/_contextlib.py", line 124, in decorate_context return func(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 2447, in _init_weights init.normal_(module.weight, mean=0.0, std=std) File "/usr/local/lib/python3.10/site-packages/transformers/initialization.py", line 54, in normal_ return TORCH_INIT_FUNCTIONS["normal_"](tensor, mean=mean, std=std, generator=generator) File "/usr/local/lib/python3.10/site-packages/torch/nn/init.py", line 265, in normal_ return _no_grad_normal_(tensor, mean, std, generator) File "/usr/local/lib/python3.10/site-packages/torch/nn/init.py", line 83, in _no_grad_normal_ return tensor.normal_(mean, std, generator=generator) NotImplementedError: "normal_kernel_cpu" not implemented for 'Byte'

Container logs: