runtime error

Exit code: 1. Reason: ormer3DModel.from_pretrained('cbensimon/Wan2.2-I2V-A14B-bf16-Diffusers', ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ subfolder='transformer_2', ^^^^^^^^^^^^^^^^^^^^^^^^^^ torch_dtype=torch.bfloat16, ^^^^^^^^^^^^^^^^^^^^^^^^^^^ device_map='cuda', ^^^^^^^^^^^^^^^^^^ ), ^ File "/root/.pyenv/versions/3.13.12/lib/python3.13/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn return fn(*args, **kwargs) File "/root/.pyenv/versions/3.13.12/lib/python3.13/site-packages/diffusers/models/modeling_utils.py", line 1288, in from_pretrained ) = cls._load_pretrained_model( ~~~~~~~~~~~~~~~~~~~~~~~~~~^ model, ^^^^^^ ...<13 lines>... is_parallel_loading_enabled=is_parallel_loading_enabled, ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ ) ^ File "/root/.pyenv/versions/3.13.12/lib/python3.13/site-packages/diffusers/models/modeling_utils.py", line 1537, in _load_pretrained_model _caching_allocator_warmup(model, expanded_device_map, dtype, hf_quantizer) ~~~~~~~~~~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/root/.pyenv/versions/3.13.12/lib/python3.13/site-packages/diffusers/models/model_loading_utils.py", line 748, in _caching_allocator_warmup _ = torch.empty(byte_count // factor, dtype=dtype, device=device, requires_grad=False) torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 26.62 GiB. GPU 0 has a total capacity of 44.39 GiB of which 17.20 GiB is free. Including non-PyTorch memory, this process has 27.18 GiB memory in use. Of the allocated memory 26.67 GiB is allocated by PyTorch, and 94.42 MiB is reserved by PyTorch but unallocated. If reserved but unallocated memory is large try setting PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True to avoid fragmentation. See documentation for Memory Management (https://pytorch.org/docs/stable/notes/cuda.html#environment-variables)

Container logs: