Set krea 2 to use new lokr format

2026-06-23 13:31:50 -06:00
parent e20b42e84a
commit 724e67d634
1 changed files with 1 additions and 0 deletions
--- a/extensions_built_in/diffusion_models/krea2/krea2.py
+++ b/extensions_built_in/diffusion_models/krea2/krea2.py
@@ -157,6 +157,7 @@ class Krea2Model(BaseModel):
        # Qwen2TokenizerFast used to tokenize the assistant suffix (matches the
        # reference's separate processor pass).
        self.processor = None
+        self.use_old_lokr_format = False

    @staticmethod
    def get_train_scheduler():