diffusers/pr_11415/en/_app/immutable/nodes/47.e1b3f592.js

import{s as ee,n as te,o as re}from"../chunks/scheduler.8c3d61f6.js";import{S as ae,i as se,g as i,s,r as T,A as ne,h as l,f as r,c as n,j as k,u as v,x as z,k as P,y as F,a,v as D,d as y,t as M,w as H}from"../chunks/index.da70eac4.js";import{D as K}from"../chunks/Docstring.fa488882.js";import{C as oe}from"../chunks/CodeBlock.a9c4becf.js";import{H as q,E as me}from"../chunks/index.dfbaf638.js";function ie(Y){let o,x,I,E,d,Z,f,A='A Transformer model for image-like data from <a href="https://huggingface.co/HiDream-ai" rel="nofollow">HiDream-I1</a>.',L,p,Q="The model can be loaded with the following code snippet.",V,u,C,c,G,g,h,R,_,W,m,$,U,w,X='The output of <a href="/docs/diffusers/pr_11415/en/api/models/transformer2d#diffusers.Transformer2DModel">Transformer2DModel</a>.',N,b,O,J,S;return d=new q({props:{title:"HiDreamImageTransformer2DModel",local:"hidreamimagetransformer2dmodel",headingTag:"h1"}}),u=new oe({props:{code:"ZnJvbSUyMGRpZmZ1c2VycyUyMGltcG9ydCUyMEhpRHJlYW1JbWFnZVRyYW5zZm9ybWVyMkRNb2RlbCUwQSUwQXRyYW5zZm9ybWVyJTIwJTNEJTIwSGlEcmVhbUltYWdlVHJhbnNmb3JtZXIyRE1vZGVsLmZyb21fcHJldHJhaW5lZCglMjJIaURyZWFtLWFpJTJGSGlEcmVhbS1JMS1GdWxsJTIyJTJDJTIwc3ViZm9sZGVyJTNEJTIydHJhbnNmb3JtZXIlMjIlMkMlMjB0b3JjaF9kdHlwZSUzRHRvcmNoLmJmbG9hdDE2KQ==",highlighted:`<span class="hljs-keyword">from</span> diffusers <span class="hljs-keyword">import</span> HiDreamImageTransformer2DModel

transformer = HiDreamImageTransformer2DModel.from_pretrained(<span class="hljs-string">&quot;HiDream-ai/HiDream-I1-Full&quot;</span>, subfolder=<span class="hljs-string">&quot;transformer&quot;</span>, torch_dtype=torch.bfloat16)`,wrap:!1}}),c=new q({props:{title:"HiDreamImageTransformer2DModel",local:"diffusers.HiDreamImageTransformer2DModel",headingTag:"h2"}}),h=new K({props:{name:"class diffusers.HiDreamImageTransformer2DModel",anchor:"diffusers.HiDreamImageTransformer2DModel",parameters:[{name:"patch_size",val:": typing.Optional[int] = None"},{name:"in_channels",val:": int = 64"},{name:"out_channels",val:": typing.Optional[int] = None"},{name:"num_layers",val:": int = 16"},{name:"num_single_layers",val:": int = 32"},{name:"attention_head_dim",val:": int = 128"},{name:"num_attention_heads",val:": int = 20"},{name:"caption_channels",val:": typing.List[int] = None"},{name:"text_emb_dim",val:": int = 2048"},{name:"num_routed_experts",val:": int = 4"},{name:"num_activated_experts",val:": int = 2"},{name:"axes_dims_rope",val:": typing.Tuple[int, int] = (32, 32)"},{name:"max_resolution",val:": typing.Tuple[int, int] = (128, 128)"},{name:"llama_layers",val:": typing.List[int] = None"},{name:"force_inference_output",val:": bool = False"}],source:"https://github.com/huggingface/diffusers/blob/vr_11415/src/diffusers/models/transformers/transformer_hidream_image.py#L605"}}),_=new q({props:{title:"Transformer2DModelOutput",local:"diffusers.models.modeling_outputs.Transformer2DModelOutput",headingTag:"h2"}}),$=new K({props:{name:"class diffusers.models.modeling_outputs.Transformer2DModelOutput",anchor:"diffusers.models.modeling_outputs.Transformer2DModelOutput",parameters:[{name:"sample",val:": torch.Tensor"}],parametersDescription:[{anchor:"diffusers.models.modeling_outputs.Transformer2DModelOutput.sample",description:`<strong>sample</strong> (<code>torch.Tensor</code> of shape <code>(batch_size, num_channels, height, width)</code> or <code>(batch size, num_vector_embeds - 1, num_latent_pixels)</code> if <a href="/docs/diffusers/pr_11415/en/api/models/transformer2d#diffusers.Transformer2DModel">Transformer2DModel</a> is discrete) &#x2014;
The hidden states output conditioned on the <code>encoder_hidden_states</code> input. If discrete, returns probability
distributions for the unnoised latent pixels.`,name:"sample"}],source:"https://github.com/huggingface/diffusers/blob/vr_11415/src/diffusers/models/modeling_outputs.py#L20"}}),b=new me({props:{source:"https://github.com/huggingface/diffusers/blob/main/docs/source/en/api/models/hidream_image_transformer.md"}}),{c(){o=i("meta"),x=s(),I=i("p"),E=s(),T(d.$$.fragment),Z=s(),f=i("p"),f.innerHTML=A,L=s(),p=i("p"),p.textContent=Q,V=s(),T(u.$$.fragment),C=s(),T(c.$$.fragment),G=s(),g=i("div"),T(h.$$.fragment),R=s(),T(_.$$.fragment),W=s(),m=i("div"),T($.$$.fragment),U=s(),w=i("p"),w.innerHTML=X,N=s(),T(b.$$.fragment),O=s(),J=i("p"),this.h()},l(e){const t=ne("svelte-u9bgzb",document.head);o=l(t,"META",{name:!0,content:!0}),t.forEach(r),x=n(e),I=l(e,"P",{}),k(I).forEach(r),E=n(e),v(d.$$.fragment,e),Z=n(e),f=l(e,"P",{"data-svelte-h":!0}),z(f)!=="svelte-ta0ebe"&&(f.innerHTML=A),L=n(e),p=l(e,"P",{"data-svelte-h":!0}),z(p)!=="svelte-1vuni30"&&(p.textContent=Q),V=n(e),v(u.$$.fragment,e),C=n(e),v(c.$$.fragment,e),G=n(e),g=l(e,"DIV",{class:!0});var B=k(g);v(h.$$.fragment,B),B.forEach(r),R=n(e),v(_.$$.fragment,e),W=n(e),m=l(e,"DIV",{class:!0});var j=k(m);v($.$$.fragment,j),U=n(j),w=l(j,"P",{"data-svelte-h":!0}),z(w)!=="svelte-1eoz4bc"&&(w.innerHTML=X),j.forEach(r),N=n(e),v(b.$$.fragment,e),O=n(e),J=l(e,"P",{}),k(J).forEach(r),this.h()},h(){P(o,"name","hf:doc:metadata"),P(o,"content",le),P(g,"class","docstring border-l-2 border-t-2 pl-4 pt-3.5 border-gray-100 rounded-tl-xl mb-6 mt-8"),P(m,"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,t){F(document.head,o),a(e,x,t),a(e,I,t),a(e,E,t),D(d,e,t),a(e,Z,t),a(e,f,t),a(e,L,t),a(e,p,t),a(e,V,t),D(u,e,t),a(e,C,t),D(c,e,t),a(e,G,t),a(e,g,t),D(h,g,null),a(e,R,t),D(_,e,t),a(e,W,t),a(e,m,t),D($,m,null),F(m,U),F(m,w),a(e,N,t),D(b,e,t),a(e,O,t),a(e,J,t),S=!0},p:te,i(e){S||(y(d.$$.fragment,e),y(u.$$.fragment,e),y(c.$$.fragment,e),y(h.$$.fragment,e),y(_.$$.fragment,e),y($.$$.fragment,e),y(b.$$.fragment,e),S=!0)},o(e){M(d.$$.fragment,e),M(u.$$.fragment,e),M(c.$$.fragment,e),M(h.$$.fragment,e),M(_.$$.fragment,e),M($.$$.fragment,e),M(b.$$.fragment,e),S=!1},d(e){e&&(r(x),r(I),r(E),r(Z),r(f),r(L),r(p),r(V),r(C),r(G),r(g),r(R),r(W),r(m),r(N),r(O),r(J)),r(o),H(d,e),H(u,e),H(c,e),H(h),H(_,e),H($),H(b,e)}}}const le='{"title":"HiDreamImageTransformer2DModel","local":"hidreamimagetransformer2dmodel","sections":[{"title":"HiDreamImageTransformer2DModel","local":"diffusers.HiDreamImageTransformer2DModel","sections":[],"depth":2},{"title":"Transformer2DModelOutput","local":"diffusers.models.modeling_outputs.Transformer2DModelOutput","sections":[],"depth":2}],"depth":1}';function de(Y){return re(()=>{new URLSearchParams(window.location.search).get("fw")}),[]}class he extends ae{constructor(o){super(),se(this,o,de,ie,ee,{})}}export{he as component};