1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

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
This commit is contained in:
Maxim Balabanski
2026-01-01 23:29:11 -08:00
committed by GitHub
parent 1cdb8723b8
commit 208cda8f6d

View File

@@ -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": {