diffusers/pr_12652/en/_app/immutable/nodes/92.2defe37b.js

import{s as le,n as fe,o as pe}from"../chunks/scheduler.53228c21.js";import{S as de,i as ge,e as o,s as n,c as _,h as ce,a as l,d as t,b as r,f as E,g as u,j as K,k,l as i,m,n as h,t as v,o as $,p as y}from"../chunks/index.100fac89.js";import{C as _e}from"../chunks/CopyLLMTxtMenu.e47a4e22.js";import{D as Q}from"../chunks/Docstring.5ca3b7d0.js";import{H as oe,E as ue}from"../chunks/MermaidChart.svelte_svelte_type_style_lang.441993b1.js";function he(ne){let f,O,H,S,b,V,T,j,x,re='A Transformer model for image-like data from <a href="https://huggingface.co/Tongyi-MAI/Z-Image-Turbo" rel="nofollow">Z-Image</a>.',q,L,A,s,D,W,p,I,X,C,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,d,M,ee,Z,ie="Patchify for basic mode: single image per batch item.",te,g,w,ae,P,me="Patchify for omni mode: multiple images per batch item with noise masks.",U,z,F,N,G;return b=new _e({props:{containerStyle:"float: right; margin-left: 10px; display: inline-flex; position: relative; z-index: 10;"}}),T=new oe({props:{title:"ZImageTransformer2DModel",local:"zimagetransformer2dmodel",headingTag:"h1"}}),L=new oe({props:{title:"ZImageTransformer2DModel",local:"diffusers.ZImageTransformer2DModel",headingTag:"h2"}}),D=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_12652/src/diffusers/models/transformers/transformer_z_image.py#L360"}}),I=new Q({props:{name:"forward",anchor:"diffusers.ZImageTransformer2DModel.forward",parameters:[{name:"x",val:": typing.Union[typing.List[torch.Tensor], typing.List[typing.List[torch.Tensor]]]"},{name:"t",val:""},{name:"cap_feats",val:": typing.Union[typing.List[torch.Tensor], typing.List[typing.List[torch.Tensor]]]"},{name:"return_dict",val:": bool = True"},{name:"controlnet_block_samples",val:": typing.Optional[typing.Dict[int, torch.Tensor]] = None"},{name:"siglip_feats",val:": typing.Optional[typing.List[typing.List[torch.Tensor]]] = None"},{name:"image_noise_mask",val:": typing.Optional[typing.List[typing.List[int]]] = None"},{name:"patch_size",val:": int = 2"},{name:"f_patch_size",val:": int = 1"}],source:"https://github.com/huggingface/diffusers/blob/vr_12652/src/diffusers/models/transformers/transformer_z_image.py#L888"}}),M=new Q({props:{name:"patchify_and_embed",anchor:"diffusers.ZImageTransformer2DModel.patchify_and_embed",parameters:[{name:"all_image",val:": typing.List[torch.Tensor]"},{name:"all_cap_feats",val:": typing.List[torch.Tensor]"},{name:"patch_size",val:": int"},{name:"f_patch_size",val:": int"}],source:"https://github.com/huggingface/diffusers/blob/vr_12652/src/diffusers/models/transformers/transformer_z_image.py#L589"}}),w=new Q({props:{name:"patchify_and_embed_omni",anchor:"diffusers.ZImageTransformer2DModel.patchify_and_embed_omni",parameters:[{name:"all_x",val:": typing.List[typing.List[torch.Tensor]]"},{name:"all_cap_feats",val:": typing.List[typing.List[torch.Tensor]]"},{name:"all_siglip_feats",val:": typing.List[typing.List[torch.Tensor]]"},{name:"patch_size",val:": int"},{name:"f_patch_size",val:": int"},{name:"images_noise_mask",val:": typing.List[typing.List[int]]"}],source:"https://github.com/huggingface/diffusers/blob/vr_12652/src/diffusers/models/transformers/transformer_z_image.py#L626"}}),z=new ue({props:{source:"https://github.com/huggingface/diffusers/blob/main/docs/source/en/api/models/z_image_transformer2d.md"}}),{c(){f=o("meta"),O=n(),H=o("p"),S=n(),_(b.$$.fragment),V=n(),_(T.$$.fragment),j=n(),x=o("p"),x.innerHTML=re,q=n(),_(L.$$.fragment),A=n(),s=o("div"),_(D.$$.fragment),W=n(),p=o("div"),_(I.$$.fragment),X=n(),C=o("p"),C.textContent=se,Y=n(),d=o("div"),_(M.$$.fragment),ee=n(),Z=o("p"),Z.textContent=ie,te=n(),g=o("div"),_(w.$$.fragment),ae=n(),P=o("p"),P.textContent=me,U=n(),_(z.$$.fragment),F=n(),N=o("p"),this.h()},l(e){const a=ce("svelte-u9bgzb",document.head);f=l(a,"META",{name:!0,content:!0}),a.forEach(t),O=r(e),H=l(e,"P",{}),E(H).forEach(t),S=r(e),u(b.$$.fragment,e),V=r(e),u(T.$$.fragment,e),j=r(e),x=l(e,"P",{"data-svelte-h":!0}),K(x)!=="svelte-1x46gmc"&&(x.innerHTML=re),q=r(e),u(L.$$.fragment,e),A=r(e),s=l(e,"DIV",{class:!0});var c=E(s);u(D.$$.fragment,c),W=r(c),p=l(c,"DIV",{class:!0});var R=E(p);u(I.$$.fragment,R),X=r(R),C=l(R,"P",{"data-svelte-h":!0}),K(C)!=="svelte-1joj8qn"&&(C.textContent=se),R.forEach(t),Y=r(c),d=l(c,"DIV",{class:!0});var B=E(d);u(M.$$.fragment,B),ee=r(B),Z=l(B,"P",{"data-svelte-h":!0}),K(Z)!=="svelte-bc1613"&&(Z.textContent=ie),B.forEach(t),te=r(c),g=l(c,"DIV",{class:!0});var J=E(g);u(w.$$.fragment,J),ae=r(J),P=l(J,"P",{"data-svelte-h":!0}),K(P)!=="svelte-1ug80iy"&&(P.textContent=me),J.forEach(t),c.forEach(t),U=r(e),u(z.$$.fragment,e),F=r(e),N=l(e,"P",{}),E(N).forEach(t),this.h()},h(){k(f,"name","hf:doc:metadata"),k(f,"content",ve),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(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(g,"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){i(document.head,f),m(e,O,a),m(e,H,a),m(e,S,a),h(b,e,a),m(e,V,a),h(T,e,a),m(e,j,a),m(e,x,a),m(e,q,a),h(L,e,a),m(e,A,a),m(e,s,a),h(D,s,null),i(s,W),i(s,p),h(I,p,null),i(p,X),i(p,C),i(s,Y),i(s,d),h(M,d,null),i(d,ee),i(d,Z),i(s,te),i(s,g),h(w,g,null),i(g,ae),i(g,P),m(e,U,a),h(z,e,a),m(e,F,a),m(e,N,a),G=!0},p:fe,i(e){G||(v(b.$$.fragment,e),v(T.$$.fragment,e),v(L.$$.fragment,e),v(D.$$.fragment,e),v(I.$$.fragment,e),v(M.$$.fragment,e),v(w.$$.fragment,e),v(z.$$.fragment,e),G=!0)},o(e){$(b.$$.fragment,e),$(T.$$.fragment,e),$(L.$$.fragment,e),$(D.$$.fragment,e),$(I.$$.fragment,e),$(M.$$.fragment,e),$(w.$$.fragment,e),$(z.$$.fragment,e),G=!1},d(e){e&&(t(O),t(H),t(S),t(V),t(j),t(x),t(q),t(A),t(s),t(U),t(F),t(N)),t(f),y(b,e),y(T,e),y(L,e),y(D),y(I),y(M),y(w),y(z,e)}}}const ve='{"title":"ZImageTransformer2DModel","local":"zimagetransformer2dmodel","sections":[{"title":"ZImageTransformer2DModel","local":"diffusers.ZImageTransformer2DModel","sections":[],"depth":2}],"depth":1}';function $e(ne){return pe(()=>{new URLSearchParams(window.location.search).get("fw")}),[]}class De extends de{constructor(f){super(),ge(this,f,$e,he,le,{})}}export{De as component};