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

@@ -307,6 +307,20 @@ export default function SimpleJob({
placeholder=""
/>
)}
{modelArch?.additionalSections?.includes('model.unconditional_lora_path') && (
<TextInput
label="Unconditional Adapter Path"
value={jobConfig.config.process[0].model.unconditional_lora_path ?? ''}
docKey="config.process[0].model.unconditional_lora_path"
onChange={(value: string | undefined) => {
if (value?.trim() === '') {
value = undefined;
}
setJobConfig(value, 'config.process[0].model.unconditional_lora_path');
}}
placeholder=""
/>
)}
{modelArch?.additionalSections?.includes('model.low_vram') && (
<FormGroup label="Options">
<Checkbox

View File

@@ -32,6 +32,7 @@ type AdditionalSections =
| 'model.low_vram'
| 'model.qie.match_target_res'
| 'model.assistant_lora_path'
| 'model.unconditional_lora_path'
| 'ideogram_4_prompt';
type ModelGroup = 'image' | 'instruction' | 'video' | 'experimental' | 'audio';
@@ -993,6 +994,10 @@ export const modelArchs: ModelArch[] = [
'config.process[0].network.conv_alpha': [undefined, 16],
'config.process[0].model.low_vram': [true, false],
'config.process[0].sample': [defaultIdeogramSamplesConfig, defaultSampleConfig],
'config.process[0].model.unconditional_lora_path': [
'ostris/ideogram_4_unconditional_lora/ideogram_4_unconditional_lora_r16.safetensors',
undefined,
],
},
disableSections: [
'network.conv',
@@ -1001,6 +1006,7 @@ export const modelArchs: ModelArch[] = [
'model.low_vram',
'model.layer_offloading',
'ideogram_4_prompt',
'model.unconditional_lora_path',
],
hasMultiLinePrompts: true,
},

View File

@@ -171,6 +171,7 @@ export interface ModelConfig {
layer_offloading_transformer_percent?: number;
layer_offloading_text_encoder_percent?: number;
assistant_lora_path?: string;
unconditional_lora_path?: string;
compile?: boolean;
block_compile?: boolean;
compile_mode?: 'default' | 'max-autotune' | 'fastest';