github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Set krea 2 to use new lokr format

Browse Source
This commit is contained in:
Jaret Burkett
2026-06-23 13:31:50 -06:00
parent e20b42e84a
commit 724e67d634
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
extensions_built_in/diffusion_models/krea2/krea2.py
View File

@@ -157,6 +157,7 @@ class Krea2Model(BaseModel):
# Qwen2TokenizerFast used to tokenize the assistant suffix (matches the # Qwen2TokenizerFast used to tokenize the assistant suffix (matches the
# reference's separate processor pass). # reference's separate processor pass).
self.processor = None self.processor = None
self.use_old_lokr_format = False
@staticmethod @staticmethod
def get_train_scheduler(): def get_train_scheduler():