runtime error

Exit code: 1. Reason: █████| 2/2 [00:07<00:00, 3.74s/it] Downloading shards: 100%|██████████| 2/2 [00:07<00:00, 3.96s/it] Traceback (most recent call last): File "/home/user/app/app.py", line 75, in <module> model = AutoModelForCausalLM.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 561, in from_pretrained return model_class.from_pretrained( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 3502, in from_pretrained ) = cls._load_pretrained_model( File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 3772, in _load_pretrained_model model.apply(model._initialize_weights) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1061, in apply module.apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1061, in apply module.apply(fn) File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1061, in apply module.apply(fn) [Previous line repeated 2 more times] File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1062, in apply fn(self) File "/usr/local/lib/python3.10/site-packages/transformers/modeling_utils.py", line 1613, in _initialize_weights self._init_weights(module) File "/usr/local/lib/python3.10/site-packages/transformers/models/phi/modeling_phi.py", line 722, in _init_weights module.weight.data.normal_(mean=0.0, std=std) File "/usr/local/lib/python3.10/site-packages/spaces/zero/torch/patching.py", line 204, in __torch_function__ res = func(*args_, **kwargs_) File "/usr/local/lib/python3.10/site-packages/spaces/zero/torch/patching.py", line 248, in __torch_dispatch__ return func(*args, **(kwargs or {})) File "/usr/local/lib/python3.10/site-packages/torch/_ops.py", line 829, in __call__ return self._op(*args, **kwargs) NotImplementedError: "normal_kernel_cpu" not implemented for 'Byte'

Container logs: