From 208cda8f6d63d2c4c855e1bdc9c3a263bbdb2f8c Mon Sep 17 00:00:00 2001
From: Maxim Balabanski <34899977+mbalabanski@users.noreply.github.com>
Date: Thu, 1 Jan 2026 23:29:11 -0800
Subject: [PATCH] fix Qwen Image Transformer single file loading mapping
 function to be consistent with other loader APIs (#12894)

fix Qwen single file loading to be consistent with other loader API
---
 src/diffusers/loaders/single_file_model.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/diffusers/loaders/single_file_model.py b/src/diffusers/loaders/single_file_model.py
index 0e4ebab7fe..f49eca3d36 100644
--- a/src/diffusers/loaders/single_file_model.py
+++ b/src/diffusers/loaders/single_file_model.py
@@ -162,7 +162,7 @@ SINGLE_FILE_LOADABLE_CLASSES = {
         "default_subfolder": "transformer",
     },
     "QwenImageTransformer2DModel": {
-        "checkpoint_mapping_fn": lambda x: x,
+        "checkpoint_mapping_fn": lambda checkpoint, **kwargs: checkpoint,
         "default_subfolder": "transformer",
     },
     "Flux2Transformer2DModel": {