runtime error

Exit code: 1. Reason: n3.10/site-packages/torch/cuda/__init__.py:734: UserWarning: Can't initialize NVML warnings.warn("Can't initialize NVML") diffusion_pytorch_model.safetensors: 0%| | 21.0M/9.63G [00:01<10:41, 15.0MB/s] diffusion_pytorch_model.safetensors: 10%|█ | 1.01G/9.63G [00:02<00:17, 503MB/s]  diffusion_pytorch_model.safetensors: 27%|██▋ | 2.62G/9.63G [00:03<00:07, 965MB/s] diffusion_pytorch_model.safetensors: 43%|████▎ | 4.10G/9.63G [00:04<00:04, 1.14GB/s] diffusion_pytorch_model.safetensors: 60%|█████▉ | 5.74G/9.63G [00:05<00:02, 1.31GB/s] diffusion_pytorch_model.safetensors: 75%|███████▍ | 7.20G/9.63G [00:06<00:01, 1.36GB/s] diffusion_pytorch_model.safetensors: 92%|█████████▏| 8.81G/9.63G [00:07<00:00, 1.43GB/s] diffusion_pytorch_model.safetensors: 100%|█████████▉| 9.63G/9.63G [00:08<00:00, 1.19GB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 61, in <module> transformer = HiDreamImageTransformer2DModel.from_pretrained( File "/usr/local/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn return fn(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/diffusers/models/modeling_utils.py", line 1218, in from_pretrained ) = cls._load_pretrained_model( File "/usr/local/lib/python3.10/site-packages/diffusers/models/modeling_utils.py", line 1477, in _load_pretrained_model offload_index, state_dict_index = load_model_dict_into_meta( File "/usr/local/lib/python3.10/site-packages/diffusers/models/model_loading_utils.py", line 285, in load_model_dict_into_meta hf_quantizer.check_quantized_param_shape(param_name, empty_state_dict[param_name], param) File "/usr/local/lib/python3.10/site-packages/diffusers/quantizers/bitsandbytes/bnb_quantizer.py", line 212, in check_quantized_param_shape n = current_param_shape.numel() AttributeError: 'tuple' object has no attribute 'numel'

Container logs: