diffusers/pr_13876/zh/_app/immutable/nodes/51.dedaec2b.js

import{s as de,n as Me,o as ce}from"../chunks/scheduler.e4ff9b64.js";import{S as ye,i as Je,e as a,s as n,c as I,h as ge,a as p,d as t,b as s,f as fe,g as W,j as o,k as re,l as he,m as i,n as $,t as v,o as _,p as E}from"../chunks/index.09f1bca0.js";import{C as we,H as be,E as Ze}from"../chunks/MermaidChart.svelte_svelte_type_style_lang.e28c9f19.js";import{C as O}from"../chunks/CodeBlock.4f2de339.js";function Ue(ee){let m,P,G,x,u,L,f,S,r,le='<a href="../api/models/auto_model">AutoPipeline</a> 是一种按<em>任务和模型</em>选择的pipeline，会根据任务自动选择正确的pipeline子类。这样你就不用提前知道具体的pipeline子类名称，也能加载不同类型的pipeline。',C,d,te="这和 <code>DiffusionPipeline</code> 不同。后者是只按<em>模型</em>选择的pipeline，会根据模型自动选择pipeline子类。",A,M,ie="<code>AutoPipelineForImage2Image</code> 会返回某个特定的pipeline子类，例如 <code>StableDiffusionXLImg2ImgPipeline</code>，它只能用于 image-to-image 任务。",k,c,H,y,ne="如果用同一个模型加载 <code>DiffusionPipeline</code>，则会返回 <code>StableDiffusionXLPipeline</code> 子类。它可以根据输入用于 text-to-image、image-to-image 或 inpainting 任务。",Y,J,q,g,se='你可以查看 <a href="https://github.com/huggingface/diffusers/blob/130fd8df54f24ffb006d84787b598d8adc899f23/src/diffusers/pipelines/auto_pipeline.py#L114" rel="nofollow">mappings</a>，确认某个模型是否受支持。',B,h,ae="如果尝试加载不受支持的模型，就会报错。",Q,w,V,b,pe='<a href="../api/models/auto_model">AutoPipeline</a> 一共有四种类型：',R,Z,oe="<li><code>AutoPipelineForText2Image</code></li> <li><code>AutoPipelineForImage2Image</code></li> <li><code>AutoPipelineForInpainting</code></li> <li><code>AutoPipelineForText2Audio</code></li>",F,U,me="这些类都带有预定义的映射关系，会把某个pipeline关联到对应任务的子类上。",z,j,ue="调用 <code>from_pretrained()</code> 时，它会从 <code>model_index.json</code> 文件中提取类名，并根据映射关系为该任务选择合适的pipeline子类。",N,T,D,X,K;return u=new we({props:{containerStyle:"float: right; margin-left: 10px; display: inline-flex; position: relative; z-index: 10;"}}),f=new be({props:{title:"AutoPipeline",local:"autopipeline",headingTag:"h1"}}),c=new O({props:{code:"aW1wb3J0JTIwdG9yY2glMEFmcm9tJTIwZGlmZnVzZXJzJTIwaW1wb3J0JTIwQXV0b1BpcGVsaW5lRm9ySW1hZ2UySW1hZ2UlMEElMEFwaXBlbGluZSUyMCUzRCUyMEF1dG9QaXBlbGluZUZvckltYWdlMkltYWdlLmZyb21fcHJldHJhaW5lZCglMEElMjAlMjAlMjJSdW5EaWZmdXNpb24lMkZKdWdnZXJuYXV0LVhMLXY5JTIyJTJDJTIwdG9yY2hfZHR5cGUlM0R0b3JjaC5iZmxvYXQxNiUyQyUyMGRldmljZV9tYXAlM0QlMjJjdWRhJTIyJTJDJTBBKSUwQXByaW50KHBpcGVsaW5lKSUwQSUyMlN0YWJsZURpZmZ1c2lvblhMSW1nMkltZ1BpcGVsaW5lJTIwJTdCJTBBJTIwJTIwJTIyX2NsYXNzX25hbWUlMjIlM0ElMjAlMjJTdGFibGVEaWZmdXNpb25YTEltZzJJbWdQaXBlbGluZSUyMiUyQyUwQSUyMCUyMC4uLiUwQSUyMg==",highlighted:`<span class="hljs-keyword">import</span> torch
<span class="hljs-keyword">from</span> diffusers <span class="hljs-keyword">import</span> AutoPipelineForImage2Image

pipeline = AutoPipelineForImage2Image.from_pretrained(
  <span class="hljs-string">&quot;RunDiffusion/Juggernaut-XL-v9&quot;</span>, torch_dtype=torch.bfloat16, device_map=<span class="hljs-string">&quot;cuda&quot;</span>,
)
<span class="hljs-built_in">print</span>(pipeline)
<span class="hljs-string">&quot;StableDiffusionXLImg2ImgPipeline {
  &quot;</span>_class_name<span class="hljs-string">&quot;: &quot;</span>StableDiffusionXLImg2ImgPipeline<span class="hljs-string">&quot;,
  ...
&quot;</span>`,lang:"py",wrap:!1}}),J=new O({props:{code:"aW1wb3J0JTIwdG9yY2glMEFmcm9tJTIwZGlmZnVzZXJzJTIwaW1wb3J0JTIwRGlmZnVzaW9uUGlwZWxpbmUlMEElMEFwaXBlbGluZSUyMCUzRCUyMERpZmZ1c2lvblBpcGVsaW5lLmZyb21fcHJldHJhaW5lZCglMEElMjAlMjAlMjJSdW5EaWZmdXNpb24lMkZKdWdnZXJuYXV0LVhMLXY5JTIyJTJDJTIwdG9yY2hfZHR5cGUlM0R0b3JjaC5iZmxvYXQxNiUyQyUyMGRldmljZV9tYXAlM0QlMjJjdWRhJTIyJTJDJTBBKSUwQXByaW50KHBpcGVsaW5lKSUwQSUyMlN0YWJsZURpZmZ1c2lvblhMUGlwZWxpbmUlMjAlN0IlMEElMjAlMjAlMjJfY2xhc3NfbmFtZSUyMiUzQSUyMCUyMlN0YWJsZURpZmZ1c2lvblhMUGlwZWxpbmUlMjIlMkMlMEElMjAlMjAuLi4lMEElMjI=",highlighted:`<span class="hljs-keyword">import</span> torch
<span class="hljs-keyword">from</span> diffusers <span class="hljs-keyword">import</span> DiffusionPipeline

pipeline = DiffusionPipeline.from_pretrained(
  <span class="hljs-string">&quot;RunDiffusion/Juggernaut-XL-v9&quot;</span>, torch_dtype=torch.bfloat16, device_map=<span class="hljs-string">&quot;cuda&quot;</span>,
)
<span class="hljs-built_in">print</span>(pipeline)
<span class="hljs-string">&quot;StableDiffusionXLPipeline {
  &quot;</span>_class_name<span class="hljs-string">&quot;: &quot;</span>StableDiffusionXLPipeline<span class="hljs-string">&quot;,
  ...
&quot;</span>`,lang:"py",wrap:!1}}),w=new O({props:{code:"aW1wb3J0JTIwdG9yY2glMEFmcm9tJTIwZGlmZnVzZXJzJTIwaW1wb3J0JTIwQXV0b1BpcGVsaW5lRm9ySW1hZ2UySW1hZ2UlMEElMEFwaXBlbGluZSUyMCUzRCUyMEF1dG9QaXBlbGluZUZvckltYWdlMkltYWdlLmZyb21fcHJldHJhaW5lZCglMEElMjAlMjAlMjAlMjAlMjJvcGVuYWklMkZzaGFwLWUtaW1nMmltZyUyMiUyQyUyMHRvcmNoX2R0eXBlJTNEdG9yY2guZmxvYXQxNiUyQyUwQSklMEElMjJWYWx1ZUVycm9yJTNBJTIwQXV0b1BpcGVsaW5lJTIwY2FuJ3QlMjBmaW5kJTIwYSUyMHBpcGVsaW5lJTIwbGlua2VkJTIwdG8lMjBTaGFwRUltZzJJbWdQaXBlbGluZSUyMGZvciUyME5vbmUlMjI=",highlighted:`<span class="hljs-keyword">import</span> torch
<span class="hljs-keyword">from</span> diffusers <span class="hljs-keyword">import</span> AutoPipelineForImage2Image

pipeline = AutoPipelineForImage2Image.from_pretrained(
    <span class="hljs-string">&quot;openai/shap-e-img2img&quot;</span>, torch_dtype=torch.float16,
)
<span class="hljs-string">&quot;ValueError: AutoPipeline can&#x27;t find a pipeline linked to ShapEImg2ImgPipeline for None&quot;</span>`,lang:"py",wrap:!1}}),T=new Ze({props:{source:"https://github.com/huggingface/diffusers/blob/main/docs/source/zh/tutorials/autopipeline.md"}}),{c(){m=a("meta"),P=n(),G=a("p"),x=n(),I(u.$$.fragment),L=n(),I(f.$$.fragment),S=n(),r=a("p"),r.innerHTML=le,C=n(),d=a("p"),d.innerHTML=te,A=n(),M=a("p"),M.innerHTML=ie,k=n(),I(c.$$.fragment),H=n(),y=a("p"),y.innerHTML=ne,Y=n(),I(J.$$.fragment),q=n(),g=a("p"),g.innerHTML=se,B=n(),h=a("p"),h.textContent=ae,Q=n(),I(w.$$.fragment),V=n(),b=a("p"),b.innerHTML=pe,R=n(),Z=a("ul"),Z.innerHTML=oe,F=n(),U=a("p"),U.textContent=me,z=n(),j=a("p"),j.innerHTML=ue,N=n(),I(T.$$.fragment),D=n(),X=a("p"),this.h()},l(e){const l=ge("svelte-u9bgzb",document.head);m=p(l,"META",{name:!0,content:!0}),l.forEach(t),P=s(e),G=p(e,"P",{}),fe(G).forEach(t),x=s(e),W(u.$$.fragment,e),L=s(e),W(f.$$.fragment,e),S=s(e),r=p(e,"P",{"data-svelte-h":!0}),o(r)!=="svelte-1wkagi6"&&(r.innerHTML=le),C=s(e),d=p(e,"P",{"data-svelte-h":!0}),o(d)!=="svelte-17fajik"&&(d.innerHTML=te),A=s(e),M=p(e,"P",{"data-svelte-h":!0}),o(M)!=="svelte-piqypq"&&(M.innerHTML=ie),k=s(e),W(c.$$.fragment,e),H=s(e),y=p(e,"P",{"data-svelte-h":!0}),o(y)!=="svelte-dvhx72"&&(y.innerHTML=ne),Y=s(e),W(J.$$.fragment,e),q=s(e),g=p(e,"P",{"data-svelte-h":!0}),o(g)!=="svelte-1pv5u6l"&&(g.innerHTML=se),B=s(e),h=p(e,"P",{"data-svelte-h":!0}),o(h)!=="svelte-oogtl0"&&(h.textContent=ae),Q=s(e),W(w.$$.fragment,e),V=s(e),b=p(e,"P",{"data-svelte-h":!0}),o(b)!=="svelte-tpgevw"&&(b.innerHTML=pe),R=s(e),Z=p(e,"UL",{"data-svelte-h":!0}),o(Z)!=="svelte-bvrics"&&(Z.innerHTML=oe),F=s(e),U=p(e,"P",{"data-svelte-h":!0}),o(U)!=="svelte-5rakjl"&&(U.textContent=me),z=s(e),j=p(e,"P",{"data-svelte-h":!0}),o(j)!=="svelte-17lnqc7"&&(j.innerHTML=ue),N=s(e),W(T.$$.fragment,e),D=s(e),X=p(e,"P",{}),fe(X).forEach(t),this.h()},h(){re(m,"name","hf:doc:metadata"),re(m,"content",je)},m(e,l){he(document.head,m),i(e,P,l),i(e,G,l),i(e,x,l),$(u,e,l),i(e,L,l),$(f,e,l),i(e,S,l),i(e,r,l),i(e,C,l),i(e,d,l),i(e,A,l),i(e,M,l),i(e,k,l),$(c,e,l),i(e,H,l),i(e,y,l),i(e,Y,l),$(J,e,l),i(e,q,l),i(e,g,l),i(e,B,l),i(e,h,l),i(e,Q,l),$(w,e,l),i(e,V,l),i(e,b,l),i(e,R,l),i(e,Z,l),i(e,F,l),i(e,U,l),i(e,z,l),i(e,j,l),i(e,N,l),$(T,e,l),i(e,D,l),i(e,X,l),K=!0},p:Me,i(e){K||(v(u.$$.fragment,e),v(f.$$.fragment,e),v(c.$$.fragment,e),v(J.$$.fragment,e),v(w.$$.fragment,e),v(T.$$.fragment,e),K=!0)},o(e){_(u.$$.fragment,e),_(f.$$.fragment,e),_(c.$$.fragment,e),_(J.$$.fragment,e),_(w.$$.fragment,e),_(T.$$.fragment,e),K=!1},d(e){e&&(t(P),t(G),t(x),t(L),t(S),t(r),t(C),t(d),t(A),t(M),t(k),t(H),t(y),t(Y),t(q),t(g),t(B),t(h),t(Q),t(V),t(b),t(R),t(Z),t(F),t(U),t(z),t(j),t(N),t(D),t(X)),t(m),E(u,e),E(f,e),E(c,e),E(J,e),E(w,e),E(T,e)}}}const je='{"title":"AutoPipeline","local":"autopipeline","sections":[],"depth":1}';function Te(ee){return ce(()=>{new URLSearchParams(window.location.search).get("fw")}),[]}class _e extends ye{constructor(m){super(),Je(this,m,Te,Ue,de,{})}}export{_e as component};