j$XddlZddlmZddlmZmZmZmZdZdZ dZ Gdd Z dS) N) build_hf_url)CLASSICEDIT STRUCTUREDSTRUCTURED_EDITzDeepBeepMeep/Flux3zDeepBeepMeep/Fibo2zQwen3-VL-4B-InstructcVeZdZedZedZedZedZedZedZ edZ edZ ed Z edd Z ed d ejfd ZedZedZedZd S)family_handlerc ddgS)N flux3_kleinflux3_klein_baserBC:\Users\johnn\OneDrive\Documents\AI\models\flux3\flux3_handler.pyquery_supported_typesz$family_handler.query_supported_types s122rcdS)Nflux3rrrrquery_model_familyz!family_handler.query_model_familyswrc ddiS)Nr)iNzFlux 3rrrrquery_family_infosz!family_handler.query_family_infoss)**rcidgdgdfS)Nr r )r r rrrrquery_family_mapsz family_handler.query_family_mapssM?I[H\]]]]rc|SNr)base_model_types r get_lora_dirzfamily_handler.get_lora_dirsrc.ddlm}|||||||S)Nr)render_prompt_helper) prompt_helperr) model_type model_def prompt_idpopup_idprompt_elem_idresolution_elem_idrs rrz#family_handler.render_prompt_helper s1777777##J 9hP^`rsssrc"ddlm}|S)Nrget_prompt_helper_css)rr(r's rr(z$family_handler.get_prompt_helper_css%s#888888$$&&&rc"ddlm}|S)Nrget_prompt_helper_javascript)rr+r*s rr+z+family_handler.get_prompt_helper_javascript*s#??????++---rc B|dk}|rdnd}iddddddgd dd dd d d |gddd|rdndddd|ddddddddddddtdd Dgd!d"d#d$dd%|rd&d'inid(|d)d*d+|d,d-gd.d/d0d1d2d.d#d3ttttddd4 S)5Nr z16 steps, no CFGz50 steps, guidance_scale=3.5 image_outputsTdevice_explicit cuda_graphs transformerskip_prompt_templatepreserve_empty_prompt_linesgroupr profiles_dirinference_stepsguidance_max_phasesrrembedded_guidanceFno_negative_promptno_background_removalvae_block_sizeresolution_budget 4096x4096 max_width max_heighttiny_vae_architectureflux2_klein_4bc Lg|]!}tttd|d"S)zQwen3-VL-4B-Instruct_z .safetensors)r ENCODER_REPO TEXT_FOLDER).0fmts r z2family_handler.query_model_def..<sD"Z"Z"Zy|,|[JsbeJsJsJs"t"t"Z"Z"Zr)bf16 int8_convrot))None)zIMain Image Ref - First Reference Sets Output Dimensions (up to 10 Images)KI)z#Ordered Reference Images (up to 10)IrMrL)choicesletters_filterdefault)text_encoder_foldertext_encoder_URLsimage_ref_choicesfit_into_canvas_image_refs acceleratednativeaeGenerate from a Text Prompt, or combine the prompt with up to 10 ordered Reference Images for edits and compositions. Main Image Ref uses the first reference to set output dimensions through the selected resolution and canvas fitting mode; later images remain additional references. Ordered Reference Images keeps the selected output dimensions. Start with u. References anchor content but do not lock composition. For a new composition, describe the complete desired image. For a bounded edit, identify the exact change and what must remain untouched. Bounding boxes in the prompt guide placement; they are not pixel masks, and edits may affect surrounding content. Prompt enhancement is optional and off by default. The upstream model supports sides from 64 to 4096 pixels, aligned to 16 pixels; start at 1024×1024. Large images and many references need more memory.uUse Prompt Helper to draw, move and resize layout boxes. For reference edits, select a Content Source and use the Source view to mark its region, then place its Output Box; removing a reference element sets a null target. The editor saves the caption and element list together, preserving comments and extra fields. For generation, write a complete scene caption: framing, subject, background, lighting and style. For a bounded edit, specify the change and retained details, e.g. "Add a wicker basket to the bicycle handlebars, matching the side-view perspective and afternoon light. Keep the bicycle, white brick wall and framing unchanged." Quote exact visible lettering. Identify references as ``, ``, etc. For layout, tag elements with `` in the caption and append a JSON array: `[{"id":"title","bbox":[80,100,220,900],"desc":"Navy title reading \"NIGHT GARDEN\""}]`. Coordinates are `[top,left,bottom,right]` on a 0–1000 grid, with the origin at top left. Reference edits use `from`, `src_bbox` and `tgt_bbox` instead of `bbox`; a null target removes an element. Review enhancer-generated coordinates: some LLMs may swap the vertical and horizontal axes. Keep caption and JSON together in one prompt. Text accuracy, exact placement and preservation are not guaranteed.aText-to-image or editing: `prompt` plus optional ordered `image_refs` (maximum 10). Use `video_prompt_type="KI"` for a main image reference: `image_refs[0]` sets output dimensions through `resolution` and `fit_canvas`; later images remain additional references. Use `video_prompt_type="I"` to keep the requested `resolution` with references, or empty for text-only generation. Defaults: u. `resolution` dimensions must be multiples of 16, each 64–4096. Bounding boxes are prompt guidance, not hard masks. No native inpainting mask workflow is declared. `prompt_enhancer=""` by default.uFor generation or a new composition, describe the finished scene. For a bounded edit, specify the changed object, concrete end state and retained details; avoid recaptioning the unchanged scene. Reference IDs are zero-based ``, etc. Quote exact visible strings. Optional layout: caption with `` tags followed by JSON elements {id,desc,bbox:[top,left,bottom,right]}, integer 0–1000 corners. Editing elements use {id,desc,from,src_bbox,tgt_bbox}; from is ref_image_0, ref_image_1, etc. or target; null tgt_bbox removes. Use multi_prompts_gen_type="FG" for a multiline caption and JSON. Review enhancer boxes for Y/X swaps; boxes guide placement, not exact pixels.)TTIT1TI1zA Complete Scene Captionz*A Detailed Image Edit using {image_inputs}z#A Scene Caption with Bounding Boxesz5A Scene Edit with Bounding Boxes using {image_inputs}) selectionlabelsrQ) infos prompt_infos deepy_infosdeepy_prompt_infosprompt_enhancer_def!text_prompt_enhancer_instructions"image_prompt_enhancer_instructions"text_prompt_enhancer_instructions1#image_prompt_enhancer_instructions1 text_prompt_enhancer_max_tokens1!image_prompt_enhancer_max_tokens1)rErrrr)rr! distilledsettingss rquery_model_defzfamily_handler.query_model_def/s*#}4 )2V%%8V T #4d z>family_handler.validate_generative_settings..fs8QQtby4D4K44"9QQQQQQrzFFlux 3 dimensions must be multiples of 16, between 64 and 4096 pixels.)lengetmapintsplitany)rr!inputswidthheights rvalidate_generative_settingsz+family_handler.validate_generative_settingsas vzz,''-2 . . 3 3GGC !5!;!;C!@!@AA v QQ%QQQ Q Q \[[ \ \rc(ddlm}|ddS)Nr)get_rgb_factorsfluxflux2) sub_family)shared.RGB_factorsr)rrs rrzfamily_handler.get_rgb_factorsis)666666v'::::rr)__name__ __module__ __qualname__ staticmethodrrrrrrr(r+rkrrtorchbfloat16r}rrrrrrr r s33\3\++\+^^\^\tt\t''\'..\.  \ >NNN\Naev{DIDR{{{\{ ==\= \\\\;;\;;;rr ) rshared.utils.hfrpromptsrrrrrprDrEr rrrrs ((((((????????????# $ a;a;a;a;a;a;a;a;a;a;r