runtime error

Exit code: 1. Reason: torch/serialization.py", line 2114, in _load result = unpickler.load() File "/usr/local/lib/python3.10/site-packages/torch/_weights_only_unpickler.py", line 532, in load self.append(self.persistent_load(pid)) File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 2078, in persistent_load typed_storage = load_tensor( File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 2044, in load_tensor wrap_storage = restore_location(storage, location) File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 1859, in restore_location return default_restore_location(storage, str(map_location)) File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 698, in default_restore_location result = fn(storage, location) File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 636, in _deserialize device = _validate_device(location, backend_name) File "/usr/local/lib/python3.10/site-packages/torch/serialization.py", line 605, in _validate_device raise RuntimeError( RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU. The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/home/user/app/app.py", line 8, in <module> model = Dia.from_pretrained("nari-labs/Dia-1.6B") File "/usr/local/lib/python3.10/site-packages/dia/model.py", line 124, in from_pretrained return cls.from_local(config_path, checkpoint_path, device) File "/usr/local/lib/python3.10/site-packages/dia/model.py", line 95, in from_local raise RuntimeError(f"Error loading checkpoint from {checkpoint_path}") from e RuntimeError: Error loading checkpoint from /home/user/.cache/huggingface/hub/models--nari-labs--Dia-1.6B/snapshots/47053d63ffae46587f294405c4fef187eeb44330/dia-v0_1.pth

Container logs: