qwen

sayakpaul · sayakpaul · commit a85f597eded1 · 2025-08-07T11:26:39.000+05:30
diff --git a/src/diffusers/loaders/single_file_model.py b/src/diffusers/loaders/single_file_model.py
@@ -153,6 +153,10 @@
         "checkpoint_mapping_fn": convert_cosmos_transformer_checkpoint_to_diffusers,
         "default_subfolder": "transformer",
     },
+    "QwenImageTransformer2DModel": {
+        "checkpoint_mapping_fn": lambda x: x,
+        "default_subfolder": "transformer",
+    },
 }
 
 

Original file line number	Diff line number	Diff line change
`@@ -153,6 +153,10 @@`
`153`	`153`	`"checkpoint_mapping_fn": convert_cosmos_transformer_checkpoint_to_diffusers,`
`154`	`154`	`"default_subfolder": "transformer",`
`155`	`155`	`},`
	`156`	`+ "QwenImageTransformer2DModel": {`
	`157`	`+ "checkpoint_mapping_fn": lambda x: x,`
	`158`	`+ "default_subfolder": "transformer",`
	`159`	`+ },`
`156`	`160`	`}`
`157`	`161`
`158`	`162`