diffusers/pr_12849/en/_app/immutable/nodes/57.e2ff02b5.js

import{s as de,n as fe,o as pe}from"../chunks/scheduler.53228c21.js";import{S as ue,i as ce,e as c,s as n,c as m,h as ge,a as g,d as a,b as s,f as P,g as i,j as B,k as A,l as te,m as r,n as o,t as l,o as d,p as f}from"../chunks/index.100fac89.js";import{C as he}from"../chunks/CopyLLMTxtMenu.733ee6d3.js";import{D as oe}from"../chunks/Docstring.695f69dc.js";import{C as le}from"../chunks/CodeBlock.d30a6509.js";import{H as K,E as $e}from"../chunks/MermaidChart.svelte_svelte_type_style_lang.0e2208d5.js";function _e(ae){let p,W,G,C,h,k,$,x,_,re='A Transformer model for image-like data from <a href="https://huggingface.co/HiDream-ai" rel="nofollow">HiDream-I1</a>.',F,y,ne="The model can be loaded with the following code snippet.",L,M,V,b,j,T,se="GGUF checkpoints for the <code>HiDreamImageTransformer2DModel</code> can  be loaded using <code>~FromOriginalModelMixin.from_single_file</code>",z,v,E,D,N,H,J,Y,U,Q,u,w,ee,Z,me='The output of <a href="/docs/diffusers/pr_12849/en/api/models/transformer2d#diffusers.Transformer2DModel">Transformer2DModel</a>.',S,R,q,I,O;return h=new he({props:{containerStyle:"float: right; margin-left: 10px; display: inline-flex; position: relative; z-index: 10;"}}),$=new K({props:{title:"HiDreamImageTransformer2DModel",local:"hidreamimagetransformer2dmodel",headingTag:"h1"}}),M=new le({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}}),b=new K({props:{title:"Loading GGUF quantized checkpoints for HiDream-I1",local:"loading-gguf-quantized-checkpoints-for-hidream-i1",headingTag:"h2"}}),v=new le({props:{code:"aW1wb3J0JTIwdG9yY2glMEFmcm9tJTIwZGlmZnVzZXJzJTIwaW1wb3J0JTIwR0dVRlF1YW50aXphdGlvbkNvbmZpZyUyQyUyMEhpRHJlYW1JbWFnZVRyYW5zZm9ybWVyMkRNb2RlbCUwQSUwQWNrcHRfcGF0aCUyMCUzRCUyMCUyMmh0dHBzJTNBJTJGJTJGaHVnZ2luZ2ZhY2UuY28lMkZjaXR5OTYlMkZIaURyZWFtLUkxLURldi1nZ3VmJTJGYmxvYiUyRm1haW4lMkZoaWRyZWFtLWkxLWRldi1RMl9LLmdndWYlMjIlMEF0cmFuc2Zvcm1lciUyMCUzRCUyMEhpRHJlYW1JbWFnZVRyYW5zZm9ybWVyMkRNb2RlbC5mcm9tX3NpbmdsZV9maWxlKCUwQSUyMCUyMCUyMCUyMGNrcHRfcGF0aCUyQyUwQSUyMCUyMCUyMCUyMHF1YW50aXphdGlvbl9jb25maWclM0RHR1VGUXVhbnRpemF0aW9uQ29uZmlnKGNvbXB1dGVfZHR5cGUlM0R0b3JjaC5iZmxvYXQxNiklMkMlMEElMjAlMjAlMjAlMjB0b3JjaF9kdHlwZSUzRHRvcmNoLmJmbG9hdDE2JTBBKQ==",highlighted:`<span class="hljs-keyword">import</span> torch
<span class="hljs-keyword">from</span> diffusers <span class="hljs-keyword">import</span> GGUFQuantizationConfig, HiDreamImageTransformer2DModel

ckpt_path = <span class="hljs-string">&quot;https://huggingface.co/city96/HiDream-I1-Dev-gguf/blob/main/hidream-i1-dev-Q2_K.gguf&quot;</span>
transformer = HiDreamImageTransformer2DModel.from_single_file(
    ckpt_path,
    quantization_config=GGUFQuantizationConfig(compute_dtype=torch.bfloat16),
    torch_dtype=torch.bfloat16
)`,wrap:!1}}),D=new K({props:{title:"HiDreamImageTransformer2DModel",local:"diffusers.HiDreamImageTransformer2DModel",headingTag:"h2"}}),J=new oe({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_12849/src/diffusers/models/transformers/transformer_hidream_image.py#L605"}}),U=new K({props:{title:"Transformer2DModelOutput",local:"diffusers.models.modeling_outputs.Transformer2DModelOutput",headingTag:"h2"}}),w=new oe({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_12849/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_12849/src/diffusers/models/modeling_outputs.py#L21"}}),R=new $e({props:{source:"https://github.com/huggingface/diffusers/blob/main/docs/source/en/api/models/hidream_image_transformer.md"}}),{c(){p=c("meta"),W=n(),G=c("p"),C=n(),m(h.$$.fragment),k=n(),m($.$$.fragment),x=n(),_=c("p"),_.innerHTML=re,F=n(),y=c("p"),y.textContent=ne,L=n(),m(M.$$.fragment),V=n(),m(b.$$.fragment),j=n(),T=c("p"),T.innerHTML=se,z=n(),m(v.$$.fragment),E=n(),m(D.$$.fragment),N=n(),H=c("div"),m(J.$$.fragment),Y=n(),m(U.$$.fragment),Q=n(),u=c("div"),m(w.$$.fragment),ee=n(),Z=c("p"),Z.innerHTML=me,S=n(),m(R.$$.fragment),q=n(),I=c("p"),this.h()},l(e){const t=ge("svelte-u9bgzb",document.head);p=g(t,"META",{name:!0,content:!0}),t.forEach(a),W=s(e),G=g(e,"P",{}),P(G).forEach(a),C=s(e),i(h.$$.fragment,e),k=s(e),i($.$$.fragment,e),x=s(e),_=g(e,"P",{"data-svelte-h":!0}),B(_)!=="svelte-ta0ebe"&&(_.innerHTML=re),F=s(e),y=g(e,"P",{"data-svelte-h":!0}),B(y)!=="svelte-1vuni30"&&(y.textContent=ne),L=s(e),i(M.$$.fragment,e),V=s(e),i(b.$$.fragment,e),j=s(e),T=g(e,"P",{"data-svelte-h":!0}),B(T)!=="svelte-p22jvc"&&(T.innerHTML=se),z=s(e),i(v.$$.fragment,e),E=s(e),i(D.$$.fragment,e),N=s(e),H=g(e,"DIV",{class:!0});var ie=P(H);i(J.$$.fragment,ie),ie.forEach(a),Y=s(e),i(U.$$.fragment,e),Q=s(e),u=g(e,"DIV",{class:!0});var X=P(u);i(w.$$.fragment,X),ee=s(X),Z=g(X,"P",{"data-svelte-h":!0}),B(Z)!=="svelte-rb9yki"&&(Z.innerHTML=me),X.forEach(a),S=s(e),i(R.$$.fragment,e),q=s(e),I=g(e,"P",{}),P(I).forEach(a),this.h()},h(){A(p,"name","hf:doc:metadata"),A(p,"content",ye),A(H,"class","docstring border-l-2 border-t-2 pl-4 pt-3.5 border-gray-100 rounded-tl-xl mb-6 mt-8"),A(u,"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){te(document.head,p),r(e,W,t),r(e,G,t),r(e,C,t),o(h,e,t),r(e,k,t),o($,e,t),r(e,x,t),r(e,_,t),r(e,F,t),r(e,y,t),r(e,L,t),o(M,e,t),r(e,V,t),o(b,e,t),r(e,j,t),r(e,T,t),r(e,z,t),o(v,e,t),r(e,E,t),o(D,e,t),r(e,N,t),r(e,H,t),o(J,H,null),r(e,Y,t),o(U,e,t),r(e,Q,t),r(e,u,t),o(w,u,null),te(u,ee),te(u,Z),r(e,S,t),o(R,e,t),r(e,q,t),r(e,I,t),O=!0},p:fe,i(e){O||(l(h.$$.fragment,e),l($.$$.fragment,e),l(M.$$.fragment,e),l(b.$$.fragment,e),l(v.$$.fragment,e),l(D.$$.fragment,e),l(J.$$.fragment,e),l(U.$$.fragment,e),l(w.$$.fragment,e),l(R.$$.fragment,e),O=!0)},o(e){d(h.$$.fragment,e),d($.$$.fragment,e),d(M.$$.fragment,e),d(b.$$.fragment,e),d(v.$$.fragment,e),d(D.$$.fragment,e),d(J.$$.fragment,e),d(U.$$.fragment,e),d(w.$$.fragment,e),d(R.$$.fragment,e),O=!1},d(e){e&&(a(W),a(G),a(C),a(k),a(x),a(_),a(F),a(y),a(L),a(V),a(j),a(T),a(z),a(E),a(N),a(H),a(Y),a(Q),a(u),a(S),a(q),a(I)),a(p),f(h,e),f($,e),f(M,e),f(b,e),f(v,e),f(D,e),f(J),f(U,e),f(w),f(R,e)}}}const ye='{"title":"HiDreamImageTransformer2DModel","local":"hidreamimagetransformer2dmodel","sections":[{"title":"Loading GGUF quantized checkpoints for HiDream-I1","local":"loading-gguf-quantized-checkpoints-for-hidream-i1","sections":[],"depth":2},{"title":"HiDreamImageTransformer2DModel","local":"diffusers.HiDreamImageTransformer2DModel","sections":[],"depth":2},{"title":"Transformer2DModelOutput","local":"diffusers.models.modeling_outputs.Transformer2DModelOutput","sections":[],"depth":2}],"depth":1}';function Me(ae){return pe(()=>{new URLSearchParams(window.location.search).get("fw")}),[]}class Ue extends ue{constructor(p){super(),ce(this,p,Me,_e,de,{})}}export{Ue as component};