runtime error

Exit code: 1. Reason: _plus.pt 0%| | 0.00/309M [00:00<?, ?B/s] 7%|▋ | 21.1M/309M [00:00<00:01, 221MB/s] 15%|█▌ | 46.6M/309M [00:00<00:01, 249MB/s] 24%|██▍ | 74.7M/309M [00:00<00:00, 269MB/s] 35%|███▍ | 108M/309M [00:00<00:00, 299MB/s] 46%|████▌ | 143M/309M [00:00<00:00, 324MB/s] 57%|█████▋ | 177M/309M [00:00<00:00, 337MB/s] 68%|██████▊ | 209M/309M [00:00<00:00, 311MB/s] 78%|███████▊ | 240M/309M [00:00<00:00, 315MB/s] 88%|████████▊ | 273M/309M [00:00<00:00, 323MB/s] 98%|█████████▊| 304M/309M [00:01<00:00, 264MB/s] 100%|██████████| 309M/309M [00:01<00:00, 289MB/s] /opt/venv/lib/python3.10/site-packages/torch/_utils.py:831: UserWarning: TypedStorage is deprecated. It will be removed in the future and UntypedStorage will be the only storage class. This should only matter to you if you are using storages directly. To access UntypedStorage directly, use tensor.untyped_storage() instead of tensor.storage() return self.fget.__get__(instance, owner)() Traceback (most recent call last): File "/app/demo_gradio.py", line 26, in <module> model, device = load_model() File "/app/demo_gradio.py", line 22, in load_model model.load_state_dict(torch.load('CNTQG_multitrain_ca44.pth', weights_only=True)['model'], strict=False) File "/opt/venv/lib/python3.10/site-packages/torch/serialization.py", line 1013, in load raise pickle.UnpicklingError(UNSAFE_MESSAGE + str(e)) from None _pickle.UnpicklingError: Weights only load failed. Re-running `torch.load` with `weights_only` set to `False` will likely succeed, but it can result in arbitrary code execution.Do it only if you get the file from a trusted source. WeightsUnpickler error: 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.

Container logs: