Add unconditional lora support so Ideogram 4 inference will more closely resemble the full pipeline results. I pushed a finetuned unconditional lora to the hub as an adapter.

This commit is contained in:
Jaret Burkett
2026-06-16 13:27:43 -06:00
parent f6c0104f25
commit 178eb5fbbe
7 changed files with 139 additions and 6 deletions

View File

@@ -620,6 +620,9 @@ class ModelConfig:
# mainly for decompression loras for distilled models
self.assistant_lora_path = kwargs.get('assistant_lora_path', None)
self.inference_lora_path = kwargs.get('inference_lora_path', None)
# a lora that stays inactive except during the unconditional (negative)
# CFG pass -- used to learn the unconditional branch without a second model
self.unconditional_lora_path = kwargs.get('unconditional_lora_path', None)
self.latent_space_version = kwargs.get('latent_space_version', None)
# only for SDXL models for now