Summary

OneTrainer exports Z-Image LoRAs with transformer.layers. key prefix instead of diffusion_model.layers.. Add this prefix (and the PEFT-wrapped base_model.model.transformer.layers. variant) to the Z-Image LoRA probe so these models are correctly identified and loaded.

QA Instructions

Import a OneTrainer Z-Image LoRA (with transformer.layers. key prefix) and verify it is detected as a Z-Image LoRA and loads correctly.

Checklist

• The PR has a short but descriptive title, suitable for a changelog
• Tests added / updated (if applicable)
• ❗Changes to a redux slice have a corresponding migration
• Documentation added / updated (if applicable)
• Updated What's New copy (if doing a release after this PR)