diff --git a/extensions_built_in/diffusion_models/krea2/krea2.py b/extensions_built_in/diffusion_models/krea2/krea2.py
index 521affd..e163820 100644
--- 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():