j6xddlZddlZddlmZddlmZmZdZdZ dZ dZ dZ d Z d Zd Zd Zd ZdZGddZdS)N) build_hf_url)query_vae_files vae_configszDeepBeepMeep/krea-2zDeepBeepMeep/Fibo2zQwen3-VL-4B-Instructz%Qwen3-VL-4B-Instruct_bf16.safetensorsz-Qwen3-VL-4B-Instruct_int8_convrot.safetensors krea2_raw krea2_turbokrea2_raw_editkrea2_turbo_editkrea2 krea2_presetsc beZdZedZedZedZedZedZedZ eddZ edddd de j e j d d dddf d Zed Zed Zed ZedZedZdZdS)family_handlerc|ttfv}gd}i|ttfvrddiniiddddgddd td d d |ttfvrd nddddddddddddddddddddddddddd gd!d"d#d$d%d&d'|d(d)d(gd*d+dt gt gtttttttttgd%dd,d-d gd.d gid/gd0 }|s| d1d2d3d4d5|r>| d6d7gd8d9d:igd;ddd?gd@dAdAdBd%d%dCd d(gd-dDg|zdd)d(gd*dE|S)FN))z)LanPaint (2 steps): ~2x slower, easy task)z+LanPaint (5 steps): ~5x slower, medium task)z+LanPaint (10 steps): ~10x slower, hard task)z0LanPaint (15 steps): ~15x slower, very hard task acceleratednativedevice_explicitT cuda_graphs transformer image_outputssystem_configstiny_vae_architecturet2vguidance_max_phasesrNAG NAG_scale?g?g{Gz?)minmaxstepNAG_taug@g? NAG_alphaginference_stepsinpaint_supportinpaint_video_prompt_typeVA inpaint_colorFFFFFFguide_custom_choices_image))zNo Control Image) Control ImageV)z#Control Image with Masked DenoisingVGr/r-r.F)choicesletters_filterdefaultlabelvisible model_modesrzInpainting Method)r1r3r4 image_modesfit_into_canvas_image_refs 2048x2048zqwen_vae_pid(1.5)z qwen_pid(1.5)) preset_profiles_dir profiles_dirtext_encoder_foldertext_encoder_URLsno_negative_promptno_background_removalresolution_budgetvae_block_size vae_upsamplervae_upsamplersexcluded_spatial_upsamplerszText-to-image from `prompt`. Inpainting combines a Control Image, mask and prompt through the selected LanPaint method; the retained image supplies context.zDescribe the finished image in natural language: subject/action, framing, setting, lighting and style. Add detail for precise composition and quote exact lettering. For inpainting, describe the desired content in the masked area within the finished scene.a'Generate an image from the Text Prompt (`prompt`). In Inpainting mode, a Control Image and mask provide the surrounding context while the prompt describes the replacement content; WanGP uses the selected LanPaint method. The text, retained context and masked region jointly determine the result.aDescribe the image you want in natural language, including subject, pose or action, framing, environment, lighting and style. Start with a short concrete idea; add detail when precise composition matters. Put literal lettering in quotes. For inpainting, describe the finished scene and the desired content inside the mask, for example: "A man in a green jacket sits at the cafe table, warm evening light.") deepy_infosdeepy_prompt_infosinfos prompt_infoszidentity-preserving editszidentity preservation)namealiasesrJzsubject placementaEdit images with the built-in Identity Edit LoRA by combining the Text Prompt (`prompt`) with Reference Images (`image_refs`). One reference supplies the image to edit; with two, use the scene first and the person or object second, and identify their roles in the prompt. Reference mode `KI` keeps the first image as the main scene; `I` treats references as people or objects. Text-only generation is also available. Inpainting combines a Control Image and mask with the prompt; choose Masked Denoising or a LanPaint method. A Control Image is placed before other references, so it becomes image 1 in the prompt. Outpainting extends the source canvas using the selected margins. Enable background removal only for references whose surrounding scene should be discarded.aUse a short, direct editing instruction and name what must remain unchanged. For example: "Change the car to matte black; keep its shape, camera angle, lighting and surroundings." For two-image composition: "Place the person from image 2 at the table in image 1; preserve their face and clothing, and match the scene lighting." Image numbers follow input order, including a Control Image placed first. State identity, pose, clothing or background constraints explicitly when they matter. For inpainting, identify the replacement inside the mask; for outpainting, describe the scene continuing into the new area. Quote exact visible text. For text-only generation, describe the finished image instead of referring to a source.a`Identity Edit combines `prompt` and ordered `image_refs`: one source, or scene first + person/object second. `KI` keeps the first image as the main scene; `I` uses people/objects. A Control Image is prepended as image 1. Inpainting uses source + mask with Masked Denoising or LanPaint; outpainting extends the canvas. Text-only generation is supported.ahGive a direct edit instruction and preservation constraints: 'Place the person from image 2 at the table in image 1; keep their face/clothing and match scene lighting.' Number inputs in order, including any Control Image first. Describe masked replacements or continuation into outpainted margins; quote exact lettering. Text-only: describe the finished image.VAG))Noner-)z[Conditional Image is first Main Subject / Landscape and may be followed by People / ObjectsKI)z'Conditional Images are People / ObjectsIrN)r1r2r3zORemove Backgrounds only behind People / Objects except main Subject / Landscape)z9Masked Denoising: inpainted area may reuse masked contentr) specialitiesrHrIrFrGr'r(image_ref_choicesat_least_one_image_ref_neededr@background_removal_labelvideo_guide_outpaintingoutpainting_quantize_marginsr6) _RAW_EDIT_MODEL_TYPE_TURBO_EDIT_MODEL_TYPE_TURBO_MODEL_TYPEr_RAW_MODEL_TYPE_PRESET_PROFILE_DIR _PROFILE_DIR_TEXT_ENCODER_FOLDERr _ENCODER_REPO_TEXT_ENCODER_BF16_FILENAME_TEXT_ENCODER_INT8_FILENAMEupdate)base_model_type model_defeditlanpaint_choicesresults BC:\Users\johnn\OneDrive\Documents\AI\models\krea2\krea2_handler.pyquery_model_defzfamily_handler.query_model_defs #79O"PP    - ,;@QSi?j,j,j x((pr- - t-  M?-  T - kmm - $U -  "OMa;b(b(b11hi-  4-  S$??-  s3==-  S$??-  t-  t-  (-  X- )|||"%( ++!- . +, !s /- < )!=- - >%8#8)N#7],@B]^^],@B]^^"#(%)!, S2QC8,;+z4family_handler.query_family_maps..|sHHHzJ HHHrhrk)rrs@rfquery_family_mapsz family_handler.query_family_mapsys4%'8:NPfg HHHHZHHHHHrhcdSNr rlrlrhrfquery_model_familyz!family_handler.query_model_family~wrhc ddiS)Nr )i~zKrea 2rlrlrhrfquery_family_infosz!family_handler.query_family_infoss)**rhcdSrvrl)ras rf get_lora_dirzfamily_handler.get_lora_dirrxrhNcLttggdgdgt|zS)N)z config.jsonztokenizer.jsonztokenizer_config.jsonzchat_template.jinjazpreprocessor_config.json)repoIdsourceFolderListfileList) _PROJECT_REPOr\r) computeListrarbs rfquery_model_filesz family_handler.query_model_filessJ(%9$:BBB    I & &' 'rhFc ddlm}|d||||| ||| |  }|j|jj|jd}t |jdr|jj|d<||fS)Nr) model_factoryckpts) checkpoint_dirmodel_filenamerqrbratext_encoder_filenamedtype VAE_dtypeVAE_upsamplingsave_quantized)r text_encodervaevisualvision_encoder) krea2_mainrrrlanguage_modelrhasattrr)rrqrarbquantizeTransformertext_encoder_quantizationrrmixed_precision_transformerrsubmodel_no_listrrkwargsrpipe_processorpipes rf load_modelzfamily_handler.load_models" .-----&")!+"7))    *5*7F!%   >. 9 9 H%3%@%GD! "t##rhcV|ttfv}|dd|rdndddd|ttfvr|ddddn3||tkrd nd |tkrdnd dd|r|d dd dSdS)Nrrrr ) image_mode batch_size model_modedenoising_strengthmasking_strength 1024x1024)num_inference_stepsguidance_scale resolution4g @rN)video_prompt_typeremove_background_images_ref)rVrWr`rX)rarb ui_defaultsrcs rfupdate_default_settingsz&family_handler.update_default_settingss2#79O"PP!1QUL\AA[\twNQRR S S S 02HI I I   qA]hii j j j j   _Pd=d=drrjlFUYmFmFmABABsvFQ R R S S S  _   T[\]] ^ ^ ^ ^ ^ _ _rhc|dd|dkr d|d<d|d<|dkr|ddkr|d|d <dSdSdS) NrrgHzG@r rrrr(r) setdefault)rasettings_versionrbrs rf fix_settingszfamily_handler.fix_settingssx|Q/// d " "03K, -.1K* + d " "{<'@A'E'E/89T/UK+ , , , # "'E'Erhc|d}d}|* t|}n#ttf$rd}YnwxYw|dvr d|d<d|d<|S)Nr)rrrrr rr)getint TypeError ValueError)inputsrmodel_mode_ints rfnormalize_lanpaint_strengthsz+family_handler.normalize_lanpaint_strengthssZZ --   ! &!$Zz* & & &!% & \ ) )+.F' (),F% &s+AAc:t|dSrj)r r)rarbrprompts rfvalidate_generative_promptz)family_handler.validate_generative_prompts33F;;;;;rhcX|ttfvrG|ddkrdnd}t|dpg|krdSt|}|dddkr|dkrt jddSdSdS) Nrrr image_refsz2Krea 2 Edit supports at most two Reference Images.rrzBDenoising Strength will be ignored if Masked Denoising is not used)rVrWrlenr rgrInfo)rarbrmax_refsrs rfvalidate_generative_settingsz+family_handler.validate_generative_settingss 35KL L L"JJ|4499qqqH6::l++1r22X==KK'DDVLL ::*A . . 2 2~7J7J GX Y Y Y Y Y 3 27J7Jrhc$ddlm}|dS)Nr)get_rgb_factorsqwen)shared.RGB_factorsr)rars rfrzfamily_handler.get_rgb_factorss$666666v&&&rhrj)__name__ __module__ __qualname__ staticmethodrgrmrtrwrzr|rtorchbfloat16float32rrrrrrrrlrhrfr r s[[\[zbb\bII\I\++\+\ ' ' '\ '!"&n-$)"%$%$%$\%$N__\_VV\V  \ <<\<ZZ\Z'''''rhr )gradiorrshared.utils.hfrmodels.qwen.vae_variantsrrrr]r\r^r_rYrXrVrWr[rZr rlrhrfrs ((((((AAAAAAAA& $ -EM!'+ %['['['['['['['['['['rh