Set ideogram to use new loke saving format

This commit is contained in:
Jaret Burkett 2026-06-07 09:49:57 -06:00
parent fa0dca288d
commit 8968e41234
1 changed files with 1 additions and 0 deletions

View File

@ -171,6 +171,7 @@ class Ideogram4Model(BaseModel):
super().__init__(
device, model_config, dtype, custom_pipeline, noise_scheduler, **kwargs
)
self.use_old_lokr_format = False
self.is_flow_matching = True
self.is_transformer = True
self.target_lora_modules = ["Ideogram4Transformer2DModel"]