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:
@@ -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
|
||||
|
||||
@@ -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,
|
||||
},
|
||||
|
||||
@@ -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';
|
||||
|
||||
Reference in New Issue
Block a user