jC DddlZddlmZddlmZmZdZdZdZdZ d Z d Z gd Z d Z d ZdZgdZdeDZddddddddddddddddddgZdZdZd dd!d"d#d$ddd%d#d&d'dd(d#d)Zd*Zd+ddddddeed, Zd'dd-Zd.Zd8d/Zd0Zd9d2Zd3Zd4Zd:d5ZGd6d7Z dS);N) build_hf_url)IDEOGRAM4_PROMPT_ENHANCERIDEOGRAM4_PROMPT_INFOSzDeepBeepMeep/Ideogram4 ideogram4zQwen3-VL-8B-InstructzDeepBeepMeep/Flux2zflux2_vae.safetensorsideogram4_turbotime))z Quality 48 V4_QUALITY_48)z Default 20 V4_DEFAULT_20)zTurbo 12 V4_TURBO_12r euler?))Eulerr )zRES 2Mres_2m)zRES 2Sres_2sch|]\}}|Sr).0_values JC:\Users\johnn\OneDrive\Documents\AI\models\ideogram4\ideogram4_handler.py rs<<<5e<<< ideogram_muz Scheduler Mufloatgg$g$@g?)idlabelnametypedefaultminmaxinc ideogram_stdz Scheduler Stdg?g?g@a## Ideogram 4 Ideogram 4 is an image model specialized for prompt-following, text rendering, layout, logos, posters, typography, and graphic design. It was trained on structured JSON captions; use that format for good results. In WanGP, optional Magic Prompt can convert a normal text brief into Ideogram's JSON caption format. Without Magic Prompt, write the structured caption yourself; directly passing plain text is not the upstream-supported prompting workflow. ## Guidance phases Ideogram 4 uses WanGP guidance phases for CFG switching. Phase 1 uses Guidance during the high-noise part of denoising. Phase 2 uses Guidance2 after the first switch threshold. If Three Phases is enabled, phase 3 uses Guidance3 after the second switch threshold. The official presets use two phases: most steps at CFG 7, then final low-noise polish at CFG 3. ## Scheduler custom settings Scheduler Mu and Scheduler Std control the Ideogram logit-normal sampling schedule. Mu shifts where denoising samples concentrate along the noise schedule. Std controls how wide that distribution is. The official presets set these values for you. Change them mainly when matching a Comfy workflow or experimenting with a different schedule. ## LoRA multipliers Ideogram 4 has two parallel transformers: the conditional transformer and the unconditional transformer. A plain LoRA multiplier applies to both branches. To set different values per branch, use ':' as conditional:unconditional. For example, 0.9:0.4 applies 0.9 to the conditional transformer and 0.4 to the unconditional transformer. Guidance phases still use ';'. For example, 1:1.2;0.8:1.0 means phase 1 cond/uncond = 1/1.2 and phase 2 cond/uncond = 0.8/1.0.a## Ideogram 4 TurboTime Ideogram 4 TurboTime uses the Ideogram 4 conditional transformer with the TurboTime LoRA applied. It is designed for a few denoising steps with no CFG and no unconditional transformer branch. LoRA multipliers apply to the conditional transformer only.0g?)num_inference_stepsmustdswitch_threshold g?i.)r r r )ideogram_cfg_override_enabled#ideogram_cfg_override_start_percent!ideogram_cfg_override_end_percent) r&guidance_phasesguidance_scaleguidance2_scaleguidance3_scaler)switch_threshold2 sample_solver flow_shiftrr#c$||zd|dz |zzz S)Nr r)shiftsigmas r_time_snr_shiftr<cs 5=C53;%"77 88rc tttt|dt|z dzS)Nr g@@)introundr<r)percentr:s r_percent_to_switch_thresholdrAgs8 u_U5\\3w3GHH6QRR S SSrcFtD]}||ddSN)_LEGACY_CFG_OVERRIDE_KEYSpop)custom_settingskeys r"_drop_legacy_cfg_override_settingsrHks4(''C&&&&''rFc|dd}t|tsi}t|tD]#\}}|r||||||<$tD]'\}}|r|dvr|||"|||<(||d<d|d<dS)NrF>r2r3r4r1r)r5 model_mode)get isinstancedictrH_TURBOTIME_CUSTOM_SETTINGSitems setdefault_TURBOTIME_SETTINGS) ui_defaultspreserve_user_valuesrFrGrs r_apply_turbotime_settingsrTps!oo&7>>O ot , ,&77706688)) U  )  & &sE 2 2 2 2#(OC )//11%% U  %C0d%d%d  " "3 . . . .$K  %4K!" $K rc Ht|}|dd}t|tsi}||d|ddt |||dddd|d t t|d dS) NrFr'r(r8r&@@r))r&r1r2r3r)r6r7rF)_PRESET_SETTINGSrKrLrMupdaterH_DEFAULT_SAMPLE_SOLVER_DEFAULT_FLOW_SHIFT)rR preset_namepresetrFs r_apply_preset_settingsr_s k *F!oo&7>>O ot , ,d|u '777%&;<"#56/)*       rcg}t|trV||dgpg||dgpgt d|DS)NURLsURLs2c3\K|]'}dt|vV(dS)_nf4N)strlower)rurls r z"_model_uses_nf4..s7::cvS)))::::::r)rLrMextendrKany) model_defurlss r_model_uses_nf4rms D)T""6 IMM&"--3444 IMM'2..4"555 ::T::: : ::rcb|tkp$t|piddS)Nconditional_transformer_onlyF)_TURBOTIME_MODEL_TYPEboolrK)base_model_typerks r_conditional_transformer_onlyrss1 3 3 ytY_"dd gtd?| dd|S)@Nz$Qwen3-VL-8B-Instruct_nf4.safetensorsz$Qwen3-VL-8B-Instruct_fp8.safetensors image_outputsTdevice_explicitflux2tiny_vae_architectureflux2_klein_4bvae_upsamplers flux2_vae_pidrexcluded_spatial_upsamplers flux2_pidguidance_max_phaseslora_multiplier_branchesconduncondinference_stepsfit_into_canvas_image_refsr profiles_dirpreset_profiles_dirideogram4_presetssample_solversr7rFc6g|]}|Sr)copy)rones r z2family_handler.query_model_def..s GGGs GGGrinfos specialitiescg|]}d|iS)rr)rrs rrz2family_handler.query_model_def..sfd^r)logosposters typographyzgraphic designztext renderingno_negative_promptno_background_removalskip_prompt_templateresolution_budget 2048x2048vae_block_sizeprompt_enhancer_button_labelz Magic Prompt prompt_infos deepy_infosa### Ideogram 4 Text-to-image for logos, posters, typography and graphic design. Trained on structured JSON captions: supply JSON, or optionally enable Magic Prompt to convert a plain-text brief. Direct plain text is not the upstream-supported workflow. Use the selected Quality/Default/Turbo preset; proofread text and layout. TurboTime has its own few-step settings. Sizes: each side 256 to 2048 pixels in multiples of 16, aspect ratio up to 6:1; 2048x2048 is the largest.deepy_prompt_infosu?### Prompt Format Supply JSON, not a bare prose prompt. Top-level order: high_level_description (recommended), style_description (optional), compositional_deconstruction (required: background, then elements). Style order: aesthetics, lighting, photo, medium for photos; aesthetics, lighting, medium, art_style otherwise; optional color_palette comes last. Use photo OR art_style. Element order: type, optional bbox, text for text elements only, desc, optional color_palette. Bboxes are [y_min,x_min,y_max,x_max], normalized 0–1000 from top left. Example: {"high_level_description":"A simple market poster.","style_description":{"aesthetics":"bold and cheerful","lighting":"flat","medium":"graphic_design","art_style":"screen print","color_palette":["#FFF4D6","#174A8B"]},"compositional_deconstruction":{"background":"Cream paper","elements":[{"type":"text","bbox":[100,100,300,900],"text":"MARKET","desc":"Large blue sans-serif lettering"}]}}. Quote every intended visible string exactly, escape embedded quotes, and close all JSON delimiters. No Markdown fences in the actual prompt.prompt_helper_popup_dimsNdprompt_enhancer_defTz&Ideogram JSON caption from Text Prompt) selectionlabelsr!text_prompt_enhancer_instructions"image_prompt_enhancer_instructionstext_prompt_enhancer_max_tokensi image_prompt_enhancer_max_tokenstext_encoder_foldertext_encoder_URLsnative)ro acceleratedrlora_multiplier_phasesrr) rsrm_SAMPLE_SOLVERS_CUSTOM_SETTINGSIDEOGRAM4_INFOSrr_TEXT_ENCODER_FOLDERr _PROJECT_REPOrZIDEOGRAM4_TURBOTIME_INFOSrE)rrrkconditional_onlytext_encoder_filenamemodel_def_updates rquery_model_defzfamily_handler.query_model_defs8)TTJYZcJdJd!Q F FkQ) T) t)  T)  $%5 ) 4 ) *K= )  "1)  '(:)  t)  )!)  _-)  "$7#8)  o)  $)  GG6FGGG) _!) " 8~#) ) $ !$%) & $T') ( #D)) * +) , b-) . +N/) 0 21) 2 z3) 4 !#f5) 6 'S 7) 8 "!U HI$$9) B 01JC) D 12KE) F .tG) H /I) J "#7K) L ],@BWXX"M) ) T  C  # #04''(*+')2 %%     !;T B B BrcdtgSNr)rprrrquery_supported_typesz$family_handler.query_supported_typess233rc iifSrCrrrrquery_family_mapsz family_handler.query_family_mapss 2v rcdSrrrrrquery_model_familyz!family_handler.query_model_family{rc ddiS)Nr)itIdeogramrrrrquery_family_infosz!family_handler.query_family_infoss/00rc.ddlm}|||||||S)Nr)render_prompt_helper) prompt_helperr) model_typerk prompt_idpopup_idprompt_elem_idresolution_elem_idrs rrz#family_handler.render_prompt_helpers1777777##J 9hP^`rsssrc"ddlm}|S)Nrget_prompt_helper_css)rrrs rrz$family_handler.get_prompt_helper_csss#888888$$&&&rc"ddlm}|S)Nrget_prompt_helper_javascript)rrrs rrz+family_handler.get_prompt_helper_javascripts#??????++---rc(ddlm}|ddS)Nr)get_rgb_factorsfluxry) sub_family)shared.RGB_factorsr)rrrs rrzfamily_handler.get_rgb_factorss)666666v'::::rcdSrr)rrs r get_lora_dirzfamily_handler.get_lora_dirrrNcdg}t||s|dtdgtggdtt t g|gdgdgS)Nz!ideogram4_transformer_config.jsonz/ideogram4_unconditional_transformer_config.jsonr)repoIdsourceFolderListfileList)z config.jsonztokenizer.jsonztokenizer_config.jsonzchat_template.jinja)rsappend _VAE_REPO _VAE_FILENAMEr_PROJECT_FOLDERr) computeListrrrktransformer_configss rquery_model_filesz family_handler.query_model_files sBC,_iHH Z  & &'X Y Y Y$%'D+_-   (%46J$K'eee     rFc ddlm} | d||||| |||  }|j|j|jd}i}|j|j|d<dgdgd}||d }||fS) Nr) model_factoryckpts) checkpoint_dirmodel_filenamerrkrrrdtype VAE_dtypesave_quantized) transformer text_encodervae transformer2r)rr)pipe coTenantsMap)ideogram4_mainrconditional_transformerr autoencoderunconditional_transformer)rrrrrkquantizeTransformertext_encoder_quantizationrrmixed_precision_transformerrsubmodel_no_listrkwargsrpipe_processorrco_tenants_maps r load_modelzfamily_handler.load_model!s 211111&")!+"7)    *A*7!-     3 ?#1#KD ./!.N *  t##rc|ddddt||rt|dSt|tdS)Nr) image_moderJ batch_size)rZrsrTr__DEFAULT_PRESET)rrrkrRs rupdate_default_settingsz&family_handler.update_default_settingsPs[!4qQQRRR () D D  %k 2 2 2 F{O<<<<rsV ((((((NNNNNNNN) - ' - "  =>SZgjsxBFOSTT O_V]jnwzDGPTUU!BD? .0s3\_``-/s4]`aa+-S[^__  +%   999TTTT''' %%%%&,;;;zzzzQ0Q0Q0Q0Q0Q0Q0Q0Q0Q0r