Set ideogram to use new loke saving format
This commit is contained in:
parent
fa0dca288d
commit
8968e41234
|
|
@ -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"]
|
||||
|
|
|
|||
Loading…
Reference in New Issue