From c5caa4fd544f842bb3f0356cf8f9344aa45624dd Mon Sep 17 00:00:00 2001
From: Maxim Balabanski <mbalabanski3@gatech.edu>
Date: Fri, 26 Dec 2025 15:46:11 -0500
Subject: [PATCH] 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 0e4ebab7fe07..f49eca3d3688 100644
--- a/src/diffusers/loaders/single_file_model.py
+++ b/src/diffusers/loaders/single_file_model.py
@@ -162,7 +162,7 @@
         "default_subfolder": "transformer",
     },
     "QwenImageTransformer2DModel": {
-        "checkpoint_mapping_fn": lambda x: x,
+        "checkpoint_mapping_fn": lambda checkpoint, **kwargs: checkpoint,
         "default_subfolder": "transformer",
     },
     "Flux2Transformer2DModel": {