Improved the prompt handeling of ideogram4 model. Now used advanced prompts class to store them smaller and allow longer prompts

2026-06-04 13:00:23 -06:00
parent 2ddc2e1318
commit 6a8d9333b6
5 changed files with 75 additions and 40 deletions
--- a/toolkit/stable_diffusion_model.py
+++ b/toolkit/stable_diffusion_model.py
@@ -269,6 +269,10 @@ class StableDiffusion:
    def is_lumina2(self):
        return self.arch == 'lumina2'
    
+    @property
+    def text_embedding_space_version(self):
+        return self.arch
+    
    @property
    def unet_unwrapped(self):
        return unwrap_model(self.unet)