diffusers/pr_13745/en/_app/immutable/nodes/68.39c3f37b.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 o,j as B,k as A,l as te,m as r,n as l,t as i,o as d,p as f}from"../chunks/index.cac5d66a.js";import{C as he}from"../chunks/CopyLLMTxtMenu.4912207d.js";import{D as le}from"../chunks/Docstring.1e7ac4f3.js";import{C as ie}from"../chunks/CodeBlock.606cbaf4.js";import{H as K,E as $e}from"../chunks/MermaidChart.svelte_svelte_type_style_lang.323ee77a.js";function _e(ae){let p,W,G,C,h,x,$,k,_,re='A Transformer model for image-like data from <a href="https://huggingface.co/HiDream-ai" rel="nofollow">HiDream-I1</a>.',F,M,ne="The model can be loaded with the following code snippet.",V,b,j,y,z,T,se="GGUF checkpoints for the <code>HiDreamImageTransformer2DModel</code> can  be loaded using <code>~FromOriginalModelMixin.from_single_file</code>",L,v,E,D,N,H,J,Y,U,Q,u,w,ee,Z,me='The output of <a href="/docs/diffusers/pr_13745/en/api/models/transformer2d#diffusers.Transformer2DModel">Transformer2DModel</a>.',S,R,q,I,X;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"}}),b=new ie({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)`,lang:"python",wrap:!1}}),y=new K({props:{title:"Loading GGUF quantized checkpoints for HiDream-I1",local:"loading-gguf-quantized-checkpoints-for-hidream-i1",headingTag:"h2"}}),v=new ie({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
)`,lang:"python",wrap:!1}}),D=new K({props:{title:"HiDreamImageTransformer2DModel",local:"diffusers.HiDreamImageTransformer2DModel",headingTag:"h2"}}),J=new le({props:{name:"class diffusers.HiDreamImageTransformer2DModel",anchor:"diffusers.HiDreamImageTransformer2DModel",parameters:[{name:"patch_size",val:": int | None = None"},{name:"in_channels",val:": int = 64"},{name:"out_channels",val:": int | None = 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:": list = 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:": tuple = (32, 32)"},{name:"max_resolution",val:": tuple = (128, 128)"},{name:"llama_layers",val:": list = None"},{name:"force_inference_output",val:": bool = False"}],source:"https://github.com/huggingface/diffusers/blob/vr_13745/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 le({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_13745/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_13745/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),x=n(),m($.$$.fragment),k=n(),_=c("p"),_.innerHTML=re,F=n(),M=c("p"),M.textContent=ne,V=n(),m(b.$$.fragment),j=n(),m(y.$$.fragment),z=n(),T=c("p"),T.innerHTML=se,L=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),o(h.$$.fragment,e),x=s(e),o($.$$.fragment,e),k=s(e),_=g(e,"P",{"data-svelte-h":!0}),B(_)!=="svelte-ta0ebe"&&(_.innerHTML=re),F=s(e),M=g(e,"P",{"data-svelte-h":!0}),B(M)!=="svelte-1vuni30"&&(M.textContent=ne),V=s(e),o(b.$$.fragment,e),j=s(e),o(y.$$.fragment,e),z=s(e),T=g(e,"P",{"data-svelte-h":!0}),B(T)!=="svelte-p22jvc"&&(T.innerHTML=se),L=s(e),o(v.$$.fragment,e),E=s(e),o(D.$$.fragment,e),N=s(e),H=g(e,"DIV",{class:!0});var oe=P(H);o(J.$$.fragment,oe),oe.forEach(a),Y=s(e),o(U.$$.fragment,e),Q=s(e),u=g(e,"DIV",{class:!0});var O=P(u);o(w.$$.fragment,O),ee=s(O),Z=g(O,"P",{"data-svelte-h":!0}),B(Z)!=="svelte-clyat2"&&(Z.innerHTML=me),O.forEach(a),S=s(e),o(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",Me),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),l(h,e,t),r(e,x,t),l($,e,t),r(e,k,t),r(e,_,t),r(e,F,t),r(e,M,t),r(e,V,t),l(b,e,t),r(e,j,t),l(y,e,t),r(e,z,t),r(e,T,t),r(e,L,t),l(v,e,t),r(e,E,t),l(D,e,t),r(e,N,t),r(e,H,t),l(J,H,null),r(e,Y,t),l(U,e,t),r(e,Q,t),r(e,u,t),l(w,u,null),te(u,ee),te(u,Z),r(e,S,t),l(R,e,t),r(e,q,t),r(e,I,t),X=!0},p:fe,i(e){X||(i(h.$$.fragment,e),i($.$$.fragment,e),i(b.$$.fragment,e),i(y.$$.fragment,e),i(v.$$.fragment,e),i(D.$$.fragment,e),i(J.$$.fragment,e),i(U.$$.fragment,e),i(w.$$.fragment,e),i(R.$$.fragment,e),X=!0)},o(e){d(h.$$.fragment,e),d($.$$.fragment,e),d(b.$$.fragment,e),d(y.$$.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),X=!1},d(e){e&&(a(W),a(G),a(C),a(x),a(k),a(_),a(F),a(M),a(V),a(j),a(z),a(T),a(L),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(b,e),f(y,e),f(v,e),f(D,e),f(J),f(U,e),f(w),f(R,e)}}}const Me='{"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 be(ae){return pe(()=>{new URLSearchParams(window.location.search).get("fw")}),[]}class Ue extends ue{constructor(p){super(),ce(this,p,be,_e,de,{})}}export{Ue as component};