model-metadata/code_execution_files

1```CODE: 2import torch3from diffusers import DiffusionPipeline4from diffusers.utils import load_image5 6# switch to "mps" for apple devices7pipe = DiffusionPipeline.from_pretrained("Qwen/Qwen-Image-Edit", dtype=torch.bfloat16, device_map="cuda")8pipe.load_lora_weights("peteromallet/Qwen-Image-Edit-InScene")9 10prompt = "Turn this cat into a dog"11input_image = load_image("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png")12 13image = pipe(image=input_image, prompt=prompt).images[0]14```15 16ERROR: 17Traceback (most recent call last):18  File "/tmp/peteromallet_Qwen-Image-Edit-InScene_0PkZxdM.py", line 28, in <module>19    pipe = DiffusionPipeline.from_pretrained("Qwen/Qwen-Image-Edit", dtype=torch.bfloat16, device_map="cuda")20  File "/tmp/.cache/uv/environments-v2/170288f40b8d9208/lib/python3.13/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn21    return fn(*args, **kwargs)22  File "/tmp/.cache/uv/environments-v2/170288f40b8d9208/lib/python3.13/site-packages/diffusers/pipelines/pipeline_utils.py", line 1025, in from_pretrained23    loaded_sub_model = load_sub_model(24        library_name=library_name,25    ...<21 lines>...26        quantization_config=quantization_config,27    )28  File "/tmp/.cache/uv/environments-v2/170288f40b8d9208/lib/python3.13/site-packages/diffusers/pipelines/pipeline_loading_utils.py", line 860, in load_sub_model29    loaded_sub_model = load_method(os.path.join(cached_folder, name), **loading_kwargs)30  File "/tmp/.cache/uv/environments-v2/170288f40b8d9208/lib/python3.13/site-packages/transformers/modeling_utils.py", line 277, in _wrapper31    return func(*args, **kwargs)32  File "/tmp/.cache/uv/environments-v2/170288f40b8d9208/lib/python3.13/site-packages/transformers/modeling_utils.py", line 5048, in from_pretrained33    ) = cls._load_pretrained_model(34        ~~~~~~~~~~~~~~~~~~~~~~~~~~^35        model,36        ^^^^^^37    ...<12 lines>...38        weights_only=weights_only,39        ^^^^^^^^^^^^^^^^^^^^^^^^^^40    )41    ^42  File "/tmp/.cache/uv/environments-v2/170288f40b8d9208/lib/python3.13/site-packages/transformers/modeling_utils.py", line 5468, in _load_pretrained_model43    _error_msgs, disk_offload_index = load_shard_file(args)44                                      ~~~~~~~~~~~~~~~^^^^^^45  File "/tmp/.cache/uv/environments-v2/170288f40b8d9208/lib/python3.13/site-packages/transformers/modeling_utils.py", line 843, in load_shard_file46    disk_offload_index = _load_state_dict_into_meta_model(47        model,48    ...<8 lines>...49        device_mesh=device_mesh,50    )51  File "/tmp/.cache/uv/environments-v2/170288f40b8d9208/lib/python3.13/site-packages/torch/utils/_contextlib.py", line 120, in decorate_context52    return func(*args, **kwargs)53  File "/tmp/.cache/uv/environments-v2/170288f40b8d9208/lib/python3.13/site-packages/transformers/modeling_utils.py", line 770, in _load_state_dict_into_meta_model54    _load_parameter_into_model(model, param_name, param.to(param_device))55                                                  ~~~~~~~~^^^^^^^^^^^^^^56torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 260.00 MiB. GPU 0 has a total capacity of 22.03 GiB of which 33.12 MiB is free. Including non-PyTorch memory, this process has 21.99 GiB memory in use. Of the allocated memory 21.79 GiB is allocated by PyTorch, and 23.18 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)57