Phase 1

2026-08-27 11:53:08 -06:00
parent e8d9cf6d35
commit 8db198ec0a
51 changed files with 985 additions and 1273 deletions
--- a/extensions_built_in/diffusion_models/ideogram4/ideogram4.py
+++ b/extensions_built_in/diffusion_models/ideogram4/ideogram4.py
@@ -26,7 +26,11 @@ from huggingface_hub.errors import EntryNotFoundError
 from transformers import AutoModel, AutoTokenizer

 from .src.transformer import Ideogram4Config, Ideogram4Transformer2DModel
-from .src.vae import AutoEncoder, AutoEncoderParams, convert_diffusers_state_dict
+from toolkit.models.v2.vae.flux2_kl import (
+    AutoEncoder,
+    AutoEncoderParams,
+    convert_diffusers_state_dict,
+)
 from .src.latent_norm import get_latent_norm
 from .src.pipeline import (
    Ideogram4Pipeline,