Added legacy paths

2026-08-28 07:55:19 -06:00
parent c3bc8b0b4e
commit 702254688d
45 changed files with 840 additions and 223 deletions
--- a/extensions_built_in/diffusion_models/z_image/z_image_l2p_model.py
+++ b/extensions_built_in/diffusion_models/z_image/z_image_l2p_model.py
@@ -22,7 +22,7 @@ HF_TOKEN = os.getenv("HF_TOKEN", None)

 try:
    from diffusers import ZImagePipeline
-    from diffusers.models.transformers.transformer_z_image import (
+    from toolkit.models.v2.diffusion_models.z_image import (
        ZImageTransformer2DModel as ZImageTransformer2DModelOriginal,
    )
    from diffusers.models.modeling_outputs import Transformer2DModelOutput