Fix random noise output by properly stripping the 'model.diffusion_model.' prefix from the AIO safetensors file before loading the state_dict.

app.py

@@ -106,7 +106,14 @@ state_dict = load_file(ckpt_path)
 config = QwenImageTransformer2DModel.load_config("Qwen/Qwen-Image-Edit-2511", subfolder="transformer")
 transformer = QwenImageTransformer2DModel.from_config(config)
 transformer.to(dtype)
-transformer.load_state_dict(state_dict, strict=False)
+
+# Extract only the transformer keys and strip the prefix
+transformer_state_dict = {}
+for k, v in state_dict.items():
+    if k.startswith("model.diffusion_model."):
+        transformer_state_dict[k.replace("model.diffusion_model.", "")] = v
+
+transformer.load_state_dict(transformer_state_dict, strict=False)
 
 # 3. Explicitly wipe the 28.4GB state_dict from CPU RAM
 del state_dict