diffusers/pr_13745/en/_app/immutable/nodes/106.020dce99.js

import{s as oe,n as fe,o as de}from"../chunks/scheduler.53228c21.js";import{S as pe,i as _e,e as i,s as r,c,h as ge,a as o,d as t,b as n,f as L,g as u,j as K,k,l as m,m as l,n as h,t as v,o as $,p as b}from"../chunks/index.cac5d66a.js";import{C as ce}from"../chunks/CopyLLMTxtMenu.4912207d.js";import{D as Q}from"../chunks/Docstring.1e7ac4f3.js";import{H as ie,E as ue}from"../chunks/MermaidChart.svelte_svelte_type_style_lang.323ee77a.js";function he(re){let f,S,N,V,x,j,y,q,T,ne='A Transformer model for image-like data from <a href="https://huggingface.co/Tongyi-MAI/Z-Image-Turbo" rel="nofollow">Z-Image</a>.',A,I,F,s,M,W,d,w,X,Z,se=`Flow: patchify -> t_embed -> x_embed -> x_refine -> cap_embed -> cap_refine
-> [siglip_embed -> siglip_refine] -> build_unified -> main_layers -> final_layer -> unpatchify`,Y,p,z,ee,P,me="Patchify for basic mode: single image per batch item.",te,_,D,ae,E,le="Patchify for omni mode: multiple images per batch item with noise masks.",G,C,O,H,R;return x=new ce({props:{containerStyle:"float: right; margin-left: 10px; display: inline-flex; position: relative; z-index: 10;"}}),y=new ie({props:{title:"ZImageTransformer2DModel",local:"zimagetransformer2dmodel",headingTag:"h1"}}),I=new ie({props:{title:"ZImageTransformer2DModel",local:"diffusers.ZImageTransformer2DModel",headingTag:"h2"}}),M=new Q({props:{name:"class diffusers.ZImageTransformer2DModel",anchor:"diffusers.ZImageTransformer2DModel",parameters:[{name:"all_patch_size",val:" = (2,)"},{name:"all_f_patch_size",val:" = (1,)"},{name:"in_channels",val:" = 16"},{name:"dim",val:" = 3840"},{name:"n_layers",val:" = 30"},{name:"n_refiner_layers",val:" = 2"},{name:"n_heads",val:" = 30"},{name:"n_kv_heads",val:" = 30"},{name:"norm_eps",val:" = 1e-05"},{name:"qk_norm",val:" = True"},{name:"cap_feat_dim",val:" = 2560"},{name:"siglip_feat_dim",val:" = None"},{name:"rope_theta",val:" = 256.0"},{name:"t_scale",val:" = 1000.0"},{name:"axes_dims",val:" = [32, 48, 48]"},{name:"axes_lens",val:" = [1024, 512, 512]"}],source:"https://github.com/huggingface/diffusers/blob/vr_13745/src/diffusers/models/transformers/transformer_z_image.py#L359"}}),w=new Q({props:{name:"forward",anchor:"diffusers.ZImageTransformer2DModel.forward",parameters:[{name:"x",val:": list"},{name:"t",val:""},{name:"cap_feats",val:": list"},{name:"return_dict",val:": bool = True"},{name:"controlnet_block_samples",val:": dict[int, torch.Tensor] | None = None"},{name:"siglip_feats",val:": list[list[torch.Tensor]] | None = None"},{name:"image_noise_mask",val:": list[list[int]] | None = None"},{name:"patch_size",val:": int = 2"},{name:"f_patch_size",val:": int = 1"}],source:"https://github.com/huggingface/diffusers/blob/vr_13745/src/diffusers/models/transformers/transformer_z_image.py#L894"}}),z=new Q({props:{name:"patchify_and_embed",anchor:"diffusers.ZImageTransformer2DModel.patchify_and_embed",parameters:[{name:"all_image",val:": list"},{name:"all_cap_feats",val:": list"},{name:"patch_size",val:": int"},{name:"f_patch_size",val:": int"}],source:"https://github.com/huggingface/diffusers/blob/vr_13745/src/diffusers/models/transformers/transformer_z_image.py#L588"}}),D=new Q({props:{name:"patchify_and_embed_omni",anchor:"diffusers.ZImageTransformer2DModel.patchify_and_embed_omni",parameters:[{name:"all_x",val:": list"},{name:"all_cap_feats",val:": list"},{name:"all_siglip_feats",val:": list"},{name:"patch_size",val:": int"},{name:"f_patch_size",val:": int"},{name:"images_noise_mask",val:": list"}],source:"https://github.com/huggingface/diffusers/blob/vr_13745/src/diffusers/models/transformers/transformer_z_image.py#L625"}}),C=new ue({props:{source:"https://github.com/huggingface/diffusers/blob/main/docs/source/en/api/models/z_image_transformer2d.md"}}),{c(){f=i("meta"),S=r(),N=i("p"),V=r(),c(x.$$.fragment),j=r(),c(y.$$.fragment),q=r(),T=i("p"),T.innerHTML=ne,A=r(),c(I.$$.fragment),F=r(),s=i("div"),c(M.$$.fragment),W=r(),d=i("div"),c(w.$$.fragment),X=r(),Z=i("p"),Z.textContent=se,Y=r(),p=i("div"),c(z.$$.fragment),ee=r(),P=i("p"),P.textContent=me,te=r(),_=i("div"),c(D.$$.fragment),ae=r(),E=i("p"),E.textContent=le,G=r(),c(C.$$.fragment),O=r(),H=i("p"),this.h()},l(e){const a=ge("svelte-u9bgzb",document.head);f=o(a,"META",{name:!0,content:!0}),a.forEach(t),S=n(e),N=o(e,"P",{}),L(N).forEach(t),V=n(e),u(x.$$.fragment,e),j=n(e),u(y.$$.fragment,e),q=n(e),T=o(e,"P",{"data-svelte-h":!0}),K(T)!=="svelte-1x46gmc"&&(T.innerHTML=ne),A=n(e),u(I.$$.fragment,e),F=n(e),s=o(e,"DIV",{class:!0});var g=L(s);u(M.$$.fragment,g),W=n(g),d=o(g,"DIV",{class:!0});var U=L(d);u(w.$$.fragment,U),X=n(U),Z=o(U,"P",{"data-svelte-h":!0}),K(Z)!=="svelte-1joj8qn"&&(Z.textContent=se),U.forEach(t),Y=n(g),p=o(g,"DIV",{class:!0});var B=L(p);u(z.$$.fragment,B),ee=n(B),P=o(B,"P",{"data-svelte-h":!0}),K(P)!=="svelte-bc1613"&&(P.textContent=me),B.forEach(t),te=n(g),_=o(g,"DIV",{class:!0});var J=L(_);u(D.$$.fragment,J),ae=n(J),E=o(J,"P",{"data-svelte-h":!0}),K(E)!=="svelte-1ug80iy"&&(E.textContent=le),J.forEach(t),g.forEach(t),G=n(e),u(C.$$.fragment,e),O=n(e),H=o(e,"P",{}),L(H).forEach(t),this.h()},h(){k(f,"name","hf:doc:metadata"),k(f,"content",ve),k(d,"class","docstring border-l-2 border-t-2 pl-4 pt-3.5 border-gray-100 rounded-tl-xl mb-6 mt-8"),k(p,"class","docstring border-l-2 border-t-2 pl-4 pt-3.5 border-gray-100 rounded-tl-xl mb-6 mt-8"),k(_,"class","docstring border-l-2 border-t-2 pl-4 pt-3.5 border-gray-100 rounded-tl-xl mb-6 mt-8"),k(s,"class","docstring border-l-2 border-t-2 pl-4 pt-3.5 border-gray-100 rounded-tl-xl mb-6 mt-8")},m(e,a){m(document.head,f),l(e,S,a),l(e,N,a),l(e,V,a),h(x,e,a),l(e,j,a),h(y,e,a),l(e,q,a),l(e,T,a),l(e,A,a),h(I,e,a),l(e,F,a),l(e,s,a),h(M,s,null),m(s,W),m(s,d),h(w,d,null),m(d,X),m(d,Z),m(s,Y),m(s,p),h(z,p,null),m(p,ee),m(p,P),m(s,te),m(s,_),h(D,_,null),m(_,ae),m(_,E),l(e,G,a),h(C,e,a),l(e,O,a),l(e,H,a),R=!0},p:fe,i(e){R||(v(x.$$.fragment,e),v(y.$$.fragment,e),v(I.$$.fragment,e),v(M.$$.fragment,e),v(w.$$.fragment,e),v(z.$$.fragment,e),v(D.$$.fragment,e),v(C.$$.fragment,e),R=!0)},o(e){$(x.$$.fragment,e),$(y.$$.fragment,e),$(I.$$.fragment,e),$(M.$$.fragment,e),$(w.$$.fragment,e),$(z.$$.fragment,e),$(D.$$.fragment,e),$(C.$$.fragment,e),R=!1},d(e){e&&(t(S),t(N),t(V),t(j),t(q),t(T),t(A),t(F),t(s),t(G),t(O),t(H)),t(f),b(x,e),b(y,e),b(I,e),b(M),b(w),b(z),b(D),b(C,e)}}}const ve='{"title":"ZImageTransformer2DModel","local":"zimagetransformer2dmodel","sections":[{"title":"ZImageTransformer2DModel","local":"diffusers.ZImageTransformer2DModel","sections":[],"depth":2}],"depth":1}';function $e(re){return de(()=>{new URLSearchParams(window.location.search).get("fw")}),[]}class Me extends pe{constructor(f){super(),_e(this,f,$e,he,oe,{})}}export{Me as component};