[bugfix] Fix QwenImageEditPipeline transformer init (#245)

Signed-off-by: dougbtv <dosmith@redhat.com>
This commit is contained in:
Doug Smith authored and GitHub committed 2025-12-09 10:11:37 +08:00
1 parent 83c2723113
commit 811b5f919d
1 file changed
+1 -1
@@ -224,7 +224,7 @@ class QwenImageEditPipeline(
self.vae = AutoencoderKLQwenImage.from_pretrained(model, subfolder="vae", local_files_only=local_files_only).to(
self.device
)
self.transformer = QwenImageTransformer2DModel()
self.transformer = QwenImageTransformer2DModel(od_config=od_config)
self.tokenizer = Qwen2Tokenizer.from_pretrained(model, subfolder="tokenizer", local_files_only=local_files_only)
self.processor = Qwen2VLProcessor.from_pretrained(
model, subfolder="processor", local_files_only=local_files_only