Buckets:

hf-doc-build
/

doc-dev

hf-doc-build/doc-dev / diffusers /pr_13881 /ja /_app /immutable /nodes /4.DbXxvkkQ.js

HuggingFaceDocBuilder's picture

HuggingFaceDocBuilder

3 days ago

32.6 kB

	import"../chunks/DsnmJJEf.js";import{i as X,h as z,C as R,H as l,a,E as Y,s as D}from"../chunks/BZtIR7es.js";import{p as x,o as E,s,f as Q,a as S,b as F,c as B,n as A}from"../chunks/_hyVs69O.js";import{D as H}from"../chunks/DbcpnoaO.js";const P='{"title":"簡単な案内","local":"簡単な案内","sections":[{"title":"拡散パイプライン","local":"拡散パイプライン","sections":[{"title":"ローカルパイプライン","local":"ローカルパイプライン","sections":[],"depth":3},{"title":"スケジューラの交換","local":"スケジューラの交換","sections":[],"depth":3}],"depth":2},{"title":"モデル","local":"モデル","sections":[],"depth":2},{"title":"スケジューラ","local":"スケジューラ","sections":[],"depth":2},{"title":"次のステップ","local":"次のステップ","sections":[],"depth":2}],"depth":1}';var L=B('<meta name="hf:doc:metadata"/>'),K=B('<p></p> <!> <!> <!> <p>拡散モデル(Diffusion Model)は、ランダムな正規分布から段階的にノイズ除去するように学習され、画像や音声などの目的のものを生成できます。これは生成AIに多大な関心を呼び起こしました。インターネット上で拡散によって生成された画像の例を見たことがあるでしょう。🧨 Diffusersは、誰もが拡散モデルに広くアクセスできるようにすることを目的としたライブラリです。</p> <p>この案内では、開発者または日常的なユーザーに関わらず、🧨 Diffusers を紹介し、素早く目的のものを生成できるようにします！このライブラリには3つの主要コンポーネントがあります:</p> <ul><li><code>DiffusionPipeline</code>は事前に学習された拡散モデルからサンプルを迅速に生成するために設計された高レベルのエンドツーエンドクラス。</li> <li>拡散システムを作成するためのビルディングブロックとして使用できる、人気のある事前学習された<a href="./api/models">モデル</a>アーキテクチャとモジュール。</li> <li>多くの異なる<a href="./api/schedulers/overview">スケジューラ</a> - ノイズがどのようにトレーニングのために加えられるか、そして生成中にどのようにノイズ除去された画像を生成するかを制御するアルゴリズム。</li></ul> <p>この案内では、<code>DiffusionPipeline</code>を生成に使用する方法を紹介し、モデルとスケジューラを組み合わせて<code>DiffusionPipeline</code>の内部で起こっていることを再現する方法を説明します。</p> <blockquote class="tip"><p>この案内は🧨 Diffusers <a href="https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers/diffusers_intro.ipynb" rel="nofollow">ノートブック</a>を簡略化したもので、すぐに使い始めることができます。Diffusers 🧨のゴール、設計哲学、コアAPIの詳細についてもっと知りたい方は、ノートブックをご覧ください！</p></blockquote> <p>始める前に必要なライブラリーがすべてインストールされていることを確認してください：</p> <!> <ul><li><a href="https://huggingface.co/docs/accelerate/index" rel="nofollow">🤗 Accelerate</a>生成とトレーニングのためのモデルのロードを高速化します</li> <li><a href="https://huggingface.co/docs/diffusers/api/pipelines/stable_diffusion/overview" rel="nofollow">Stable Diffusion</a>ような最も一般的な拡散モデルを実行するには、<a href="https://huggingface.co/docs/transformers/index" rel="nofollow">🤗 Transformers</a>が必要です。</li></ul> <!> <p><code>DiffusionPipeline</code>は事前学習された拡散システムを生成に使用する最も簡単な方法です。これはモデルとスケジューラを含むエンドツーエンドのシステムです。<code>DiffusionPipeline</code>は多くの作業／タスクにすぐに使用することができます。また、サポートされているタスクの完全なリストについては<a href="./api/pipelines/overview#diffusers-summary">🧨Diffusersの概要</a>の表を参照してください。</p> <table><thead><tr><th><strong>タスク</strong></th><th><strong>説明</strong></th><th><strong>パイプライン</strong></th></tr></thead><tbody><tr><td>Unconditional Image Generation</td><td>正規分布から画像生成</td><td><a href="./using-diffusers/unconditional_image_generation">unconditional_image_generation</a></td></tr><tr><td>Text-Guided Image Generation</td><td>文章から画像生成</td><td><a href="./using-diffusers/conditional_image_generation">conditional_image_generation</a></td></tr><tr><td>Text-Guided Image-to-Image Translation</td><td>画像と文章から新たな画像生成</td><td><a href="./using-diffusers/img2img">img2img</a></td></tr><tr><td>Text-Guided Image-Inpainting</td><td>画像、マスク、および文章が指定された場合に、画像のマスクされた部分を文章をもとに修復</td><td><a href="./using-diffusers/inpaint">inpaint</a></td></tr><tr><td>Text-Guided Depth-to-Image Translation</td><td>文章と深度推定によって構造を保持しながら画像生成</td><td><a href="./using-diffusers/depth2img">depth2img</a></td></tr></tbody></table> <p>まず、<code>DiffusionPipeline</code>のインスタンスを作成し、ダウンロードしたいパイプラインのチェックポイントを指定します。\nこの<code>DiffusionPipeline</code>はHugging Face Hubに保存されている任意の<a href="https://huggingface.co/models?library=diffusers&sort=downloads" rel="nofollow">チェックポイント</a>を使用することができます。\nこの案内では、<a href="https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5" rel="nofollow"><code>stable-diffusion-v1-5</code></a>チェックポイントでテキストから画像へ生成します。</p> <blockquote><p>[!WARNING][Stable Diffusion]モデルについては、モデルを実行する前にまず<a href="https://huggingface.co/spaces/CompVis/stable-diffusion-license" rel="nofollow">ライセンス</a>を注意深くお読みください。🧨 Diffusers は、攻撃的または有害なコンテンツを防ぐために <a href="https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/stable_diffusion/safety_checker.py" rel="nofollow"><code>safety_checker</code></a> を実装していますが、モデルの改良された画像生成機能により、潜在的に有害なコンテンツが生成される可能性があります。</p></blockquote> <p>モデルを<code>from_pretrained()</code>メソッドでロードします：</p> <!> <p><code>DiffusionPipeline</code>は全てのモデリング、トークン化、スケジューリングコンポーネントをダウンロードしてキャッシュします。Stable Diffusionパイプラインは<code>UNet2DConditionModel</code>と<code>PNDMScheduler</code>などで構成されています：</p> <!> <p>このモデルはおよそ14億個のパラメータで構成されているため、GPU上でパイプラインを実行することを強く推奨します。\nPyTorchと同じように、ジェネレータオブジェクトをGPUに移すことができます：</p> <!> <p>これで、文章を <code>pipeline</code> に渡して画像を生成し、ノイズ除去された画像にアクセスできるようになりました。デフォルトでは、画像出力は<a href="https://pillow.readthedocs.io/en/stable/reference/Image.html?highlight=image#the-image-class" rel="nofollow"><code>PIL.Image</code></a>オブジェクトでラップされます。</p> <!> <div class="flex justify-center"><img src="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/image_of_squirrel_painting.png"/></div> <p><code>save</code>関数で画像を保存できます:</p> <!> <!> <p>ローカルでパイプラインを使用することもできます。唯一の違いは、最初にウェイトをダウンロードする必要があることです：</p> <!> <p>保存したウェイトをパイプラインにロードします：</p> <!> <p>これで、上のセクションと同じようにパイプラインを動かすことができます。</p> <!> <p>スケジューラーによって、ノイズ除去のスピードや品質のトレードオフが異なります。どれが自分に最適かを知る最善の方法は、実際に試してみることです！Diffusers 🧨の主な機能の1つは、スケジューラを簡単に切り替えることができることです。例えば、デフォルトの<code>PNDMScheduler</code>を<code>EulerDiscreteScheduler</code>に置き換えるには、<code>from_config()</code>メソッドでロードできます：</p> <!> <p>新しいスケジューラを使って画像を生成し、その違いに気づくかどうか試してみてください！</p> <p>次のセクションでは、<code>DiffusionPipeline</code>を構成するコンポーネント（モデルとスケジューラ）を詳しく見て、これらのコンポーネントを使って猫の画像を生成する方法を学びます。</p> <!> <p>ほとんどのモデルはノイズの多いサンプルを取り、各タイムステップで<em>残りのノイズ</em>を予測します（他のモデルは前のサンプルを直接予測するか、速度または<a href="https://github.com/huggingface/diffusers/blob/5e5ce13e2f89ac45a0066cb3f369462a3cf1d9ef/src/diffusers/schedulers/scheduling_ddim.py#L110" rel="nofollow"><code>v-prediction</code></a>を予測するように学習します）。モデルを混ぜて他の拡散システムを作ることもできます。</p> <p>モデルは<code>from_pretrained()</code>メソッドで開始されます。このメソッドはモデルをローカルにキャッシュするので、次にモデルをロードするときに高速になります。この案内では、<code>UNet2DModel</code>をロードします。これは基本的な画像生成モデルであり、猫画像で学習されたチェックポイントを使います：</p> <!> <p>モデルのパラメータにアクセスするには、<code>model.config</code> を呼び出せます：</p> <!> <p>モデル構成は🧊凍結🧊されたディクショナリであり、モデル作成後にこれらのパラメータを変更することはできません。これは意図的なもので、最初にモデル・アーキテクチャを定義するために使用されるパラメータが同じままであることを保証します。他のパラメータは生成中に調整することができます。</p> <p>最も重要なパラメータは以下の通りです：</p> <ul><li>sample_size`: 入力サンプルの高さと幅。</li> <li><code>in_channels</code>: 入力サンプルの入力チャンネル数。</li> <li>down_block_types<code>と</code>up_block_types`: UNet アーキテクチャを作成するために使用されるダウンサンプリングブロックとアップサンプリングブロックのタイプ。</li> <li>block_out_channels`: ダウンサンプリングブロックの出力チャンネル数。逆順でアップサンプリングブロックの入力チャンネル数にも使用されます。</li> <li>layer_per_block`: 各 UNet ブロックに含まれる ResNet ブロックの数。</li></ul> <p>このモデルを生成に使用するには、ランダムな画像の形の正規分布を作成します。このモデルは複数のランダムな正規分布を受け取ることができるため<code>batch</code>軸を入れます。入力チャンネル数に対応する<code>channel</code>軸も必要です。画像の高さと幅に対応する<code>sample_size</code>軸を持つ必要があります：</p> <!> <p>画像生成には、ノイズの多い画像と <code>timestep</code> をモデルに渡します。<code>timestep</code>は入力画像がどの程度ノイズが多いかを示します。これは、モデルが拡散プロセスにおける自分の位置を決定するのに役立ちます。モデルの出力を得るには <code>sample</code> メソッドを使用します：</p> <!> <p>しかし、実際の例を生成するには、ノイズ除去プロセスをガイドするスケジューラが必要です。次のセクションでは、モデルをスケジューラと組み合わせる方法を学びます。</p> <!> <p>スケジューラは、モデルの出力（この場合は <code>noisy_residual</code> ）が与えられたときに、ノイズの多いサンプルからノイズの少ないサンプルへの移行を管理します。</p> <blockquote class="tip"><p>🧨 Diffusersは拡散システムを構築するためのツールボックスです。<code>DiffusionPipeline</code>は事前に構築された拡散システムを使い始めるのに便利な方法ですが、独自のモデルとスケジューラコンポーネントを個別に選択してカスタム拡散システムを構築することもできます。</p></blockquote> <p>この案内では、<code>DDPMScheduler</code>を<code>from_config()</code>メソッドでインスタンス化します：</p> <!> <blockquote class="tip"><p>💡 スケジューラがどのようにコンフィギュレーションからインスタンス化されるかに注目してください。モデルとは異なり、スケジューラは学習可能な重みを持たず、パラメーターを持ちません！</p></blockquote> <p>最も重要なパラメータは以下の通りです：</p> <ul><li>num_train_timesteps`: ノイズ除去処理の長さ、言い換えれば、ランダムな正規分布をデータサンプルに処理するのに必要なタイムステップ数です。</li> <li><code>beta_schedule</code>: 生成とトレーニングに使用するノイズスケジュールのタイプ。</li> <li><code>beta_start</code> と <code>beta_end</code>: ノイズスケジュールの開始値と終了値。</li></ul> <p>少しノイズの少ない画像を予測するには、スケジューラの <code>step()</code> メソッドに以下を渡します: モデルの出力、<code>timestep</code>、現在の <code>sample</code>。</p> <!> <p><code>less_noisy_sample</code>は次の<code>timestep</code>に渡すことができ、そこでさらにノイズが少なくなります！</p> <p>では、すべてをまとめて、ノイズ除去プロセス全体を視覚化してみましょう。</p> <p>まず、ノイズ除去された画像を後処理して <code>PIL.Image</code> として表示する関数を作成します：</p> <!> <p>ノイズ除去処理を高速化するために入力とモデルをGPUに移します：</p> <!> <p>ここで、ノイズが少なくなったサンプルの残りのノイズを予測するノイズ除去ループを作成し、スケジューラを使ってさらにノイズの少ないサンプルを計算します：</p> <!> <p>何もないところから猫が生成されるのを、座って見てください！😻</p> <div class="flex justify-center"><img src="https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/diffusion-quicktour.png"/></div> <!> <p>このクイックツアーで、🧨ディフューザーを使ったクールな画像をいくつか作成できたと思います！次のステップとして</p> <ul><li>モデルをトレーニングまたは微調整については、<a href="./tutorials/basic_training">training</a>チュートリアルを参照してください。</li> <li>様々な使用例については、公式およびコミュニティの<a href="https://github.com/huggingface/diffusers/tree/main/examples#-diffusers-examples" rel="nofollow">training or finetuning scripts</a>の例を参照してください。</li> <li>スケジューラのロード、アクセス、変更、比較については<a href="./using-diffusers/schedulers">Using different Schedulers</a>ガイドを参照してください。</li> <li>プロンプトエンジニアリング、スピードとメモリの最適化、より高品質な画像を生成するためのヒントやトリックについては、<a href="./stable_diffusion">Stable Diffusion</a>ガイドを参照してください。</li> <li>🧨 Diffusers の高速化については、最適化された <a href="./optimization/fp16">PyTorch on a GPU</a>のガイド、<a href="./optimization/mps">Stable Diffusion on Apple Silicon (M1/M2)</a>と<a href="./optimization/onnx">ONNX Runtime</a>を参照してください。</li></ul> <!> <p></p>',1);function ls(C,V){x(V,!1),E(()=>{new URLSearchParams(window.location.search).get("fw")}),X();var e=K();z("1at4a0u",q=>{var I=L();D(I,"content",P),S(q,I)});var n=s(Q(e),2);R(n,{containerStyle:"float: right; margin-left: 10px; display: inline-flex; position: relative; z-index: 10;"});var t=s(n,2);H(t,{containerStyle:"float: right; margin-left: 10px; display: inline-flex; position: relative; z-index: 10;",options:[{label:"Mixed",value:"https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers_doc/ja/quicktour.ipynb"},{label:"PyTorch",value:"https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers_doc/ja/pytorch/quicktour.ipynb"},{label:"TensorFlow",value:"https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers_doc/ja/tensorflow/quicktour.ipynb"},{label:"Mixed",value:"https://studiolab.sagemaker.aws/import/github/huggingface/notebooks/blob/main/diffusers_doc/ja/quicktour.ipynb"},{label:"PyTorch",value:"https://studiolab.sagemaker.aws/import/github/huggingface/notebooks/blob/main/diffusers_doc/ja/pytorch/quicktour.ipynb"},{label:"TensorFlow",value:"https://studiolab.sagemaker.aws/import/github/huggingface/notebooks/blob/main/diffusers_doc/ja/tensorflow/quicktour.ipynb"}]});var p=s(t,2);l(p,{title:"簡単な案内",local:"簡単な案内",headingTag:"h1"});var i=s(p,14);a(i,{code:"JTIzJTIwdW5jb21tZW50JTIwdG8lMjBpbnN0YWxsJTIwdGhlJTIwbmVjZXNzYXJ5JTIwbGlicmFyaWVzJTIwaW4lMjBDb2xhYiUwQSUyMyFwaXAlMjBpbnN0YWxsJTIwLS11cGdyYWRlJTIwZGlmZnVzZXJzJTIwYWNjZWxlcmF0ZSUyMHRyYW5zZm9ybWVycw==",highlighted:`<span class="hljs-comment"># uncomment to install the necessary libraries in Colab</span>
	<span class="hljs-comment">#!pip install --upgrade diffusers accelerate transformers</span>`,lang:"py",wrap:!1});var o=s(i,4);l(o,{title:"拡散パイプライン",local:"拡散パイプライン",headingTag:"h2"});var c=s(o,12);a(c,{code:"ZnJvbSUyMGRpZmZ1c2VycyUyMGltcG9ydCUyMERpZmZ1c2lvblBpcGVsaW5lJTBBJTBBcGlwZWxpbmUlMjAlM0QlMjBEaWZmdXNpb25QaXBlbGluZS5mcm9tX3ByZXRyYWluZWQoJTIyc3RhYmxlLWRpZmZ1c2lvbi12MS01JTJGc3RhYmxlLWRpZmZ1c2lvbi12MS01JTIyJTJDJTIwdXNlX3NhZmV0ZW5zb3JzJTNEVHJ1ZSk=",highlighted:`<span class="hljs-meta">>>> </span><span class="hljs-keyword">from</span> diffusers <span class="hljs-keyword">import</span> DiffusionPipeline

	<span class="hljs-meta">>>> </span>pipeline = DiffusionPipeline.from_pretrained(<span class="hljs-string">"stable-diffusion-v1-5/stable-diffusion-v1-5"</span>, use_safetensors=<span class="hljs-literal">True</span>)`,lang:"python",wrap:!1});var d=s(c,4);a(d,{code:"cGlwZWxpbmU=",highlighted:`<span class="hljs-meta">>>> </span>pipeline
	StableDiffusionPipeline {
	<span class="hljs-string">"_class_name"</span>: <span class="hljs-string">"StableDiffusionPipeline"</span>,
	<span class="hljs-string">"_diffusers_version"</span>: <span class="hljs-string">"0.13.1"</span>,
	...,
	<span class="hljs-string">"scheduler"</span>: [
	<span class="hljs-string">"diffusers"</span>,
	<span class="hljs-string">"PNDMScheduler"</span>
	],
	...,
	<span class="hljs-string">"unet"</span>: [
	<span class="hljs-string">"diffusers"</span>,
	<span class="hljs-string">"UNet2DConditionModel"</span>
	],
	<span class="hljs-string">"vae"</span>: [
	<span class="hljs-string">"diffusers"</span>,
	<span class="hljs-string">"AutoencoderKL"</span>
	]
	}`,lang:"py",wrap:!1});var r=s(d,4);a(r,{code:"cGlwZWxpbmUudG8oJTIyY3VkYSUyMik=",highlighted:'<span class="hljs-meta">>>> </span>pipeline.to(<span class="hljs-string">"cuda"</span>)',lang:"python",wrap:!1});var h=s(r,4);a(h,{code:"aW1hZ2UlMjAlM0QlMjBwaXBlbGluZSglMjJBbiUyMGltYWdlJTIwb2YlMjBhJTIwc3F1aXJyZWwlMjBpbiUyMFBpY2Fzc28lMjBzdHlsZSUyMikuaW1hZ2VzJTVCMCU1RCUwQWltYWdl",highlighted:`<span class="hljs-meta">>>> </span>image = pipeline(<span class="hljs-string">"An image of a squirrel in Picasso style"</span>).images[<span class="hljs-number">0</span>]
	<span class="hljs-meta">>>> </span>image`,lang:"python",wrap:!1});var g=s(h,6);a(g,{code:"aW1hZ2Uuc2F2ZSglMjJpbWFnZV9vZl9zcXVpcnJlbF9wYWludGluZy5wbmclMjIp",highlighted:'<span class="hljs-meta">>>> </span>image.save(<span class="hljs-string">"image_of_squirrel_painting.png"</span>)',lang:"python",wrap:!1});var u=s(g,2);l(u,{title:"ローカルパイプライン",local:"ローカルパイプライン",headingTag:"h3"});var m=s(u,4);a(m,{code:"IWdpdCUyMGxmcyUyMGluc3RhbGwlMEEhZ2l0JTIwY2xvbmUlMjBodHRwcyUzQSUyRiUyRmh1Z2dpbmdmYWNlLmNvJTJGc3RhYmxlLWRpZmZ1c2lvbi12MS01JTJGc3RhYmxlLWRpZmZ1c2lvbi12MS01",highlighted:`!git lfs install
	!git <span class="hljs-built_in">clone</span> https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5`,lang:"bash",wrap:!1});var f=s(m,4);a(f,{code:"cGlwZWxpbmUlMjAlM0QlMjBEaWZmdXNpb25QaXBlbGluZS5mcm9tX3ByZXRyYWluZWQoJTIyLiUyRnN0YWJsZS1kaWZmdXNpb24tdjEtNSUyMiUyQyUyMHVzZV9zYWZldGVuc29ycyUzRFRydWUp",highlighted:'<span class="hljs-meta">>>> </span>pipeline = DiffusionPipeline.from_pretrained(<span class="hljs-string">"./stable-diffusion-v1-5"</span>, use_safetensors=<span class="hljs-literal">True</span>)',lang:"python",wrap:!1});var b=s(f,4);l(b,{title:"スケジューラの交換",local:"スケジューラの交換",headingTag:"h3"});var y=s(b,4);a(y,{code:"ZnJvbSUyMGRpZmZ1c2VycyUyMGltcG9ydCUyMEV1bGVyRGlzY3JldGVTY2hlZHVsZXIlMEElMEFwaXBlbGluZSUyMCUzRCUyMERpZmZ1c2lvblBpcGVsaW5lLmZyb21fcHJldHJhaW5lZCglMjJzdGFibGUtZGlmZnVzaW9uLXYxLTUlMkZzdGFibGUtZGlmZnVzaW9uLXYxLTUlMjIlMkMlMjB1c2Vfc2FmZXRlbnNvcnMlM0RUcnVlKSUwQXBpcGVsaW5lLnNjaGVkdWxlciUyMCUzRCUyMEV1bGVyRGlzY3JldGVTY2hlZHVsZXIuZnJvbV9jb25maWcocGlwZWxpbmUuc2NoZWR1bGVyLmNvbmZpZyk=",highlighted:`<span class="hljs-meta">>>> </span><span class="hljs-keyword">from</span> diffusers <span class="hljs-keyword">import</span> EulerDiscreteScheduler

	<span class="hljs-meta">>>> </span>pipeline = DiffusionPipeline.from_pretrained(<span class="hljs-string">"stable-diffusion-v1-5/stable-diffusion-v1-5"</span>, use_safetensors=<span class="hljs-literal">True</span>)
	<span class="hljs-meta">>>> </span>pipeline.scheduler = EulerDiscreteScheduler.from_config(pipeline.scheduler.config)`,lang:"py",wrap:!1});var M=s(y,6);l(M,{title:"モデル",local:"モデル",headingTag:"h2"});var j=s(M,6);a(j,{code:"ZnJvbSUyMGRpZmZ1c2VycyUyMGltcG9ydCUyMFVOZXQyRE1vZGVsJTBBJTBBcmVwb19pZCUyMCUzRCUyMCUyMmdvb2dsZSUyRmRkcG0tY2F0LTI1NiUyMiUwQW1vZGVsJTIwJTNEJTIwVU5ldDJETW9kZWwuZnJvbV9wcmV0cmFpbmVkKHJlcG9faWQlMkMlMjB1c2Vfc2FmZXRlbnNvcnMlM0RUcnVlKQ==",highlighted:`<span class="hljs-meta">>>> </span><span class="hljs-keyword">from</span> diffusers <span class="hljs-keyword">import</span> UNet2DModel

	<span class="hljs-meta">>>> </span>repo_id = <span class="hljs-string">"google/ddpm-cat-256"</span>
	<span class="hljs-meta">>>> </span>model = UNet2DModel.from_pretrained(repo_id, use_safetensors=<span class="hljs-literal">True</span>)`,lang:"py",wrap:!1});var w=s(j,4);a(w,{code:"bW9kZWwuY29uZmln",highlighted:'<span class="hljs-meta">>>> </span>model.config',lang:"py",wrap:!1});var U=s(w,10);a(U,{code:"aW1wb3J0JTIwdG9yY2glMEElMEF0b3JjaC5tYW51YWxfc2VlZCgwKSUwQSUwQW5vaXN5X3NhbXBsZSUyMCUzRCUyMHRvcmNoLnJhbmRuKDElMkMlMjBtb2RlbC5jb25maWcuaW5fY2hhbm5lbHMlMkMlMjBtb2RlbC5jb25maWcuc2FtcGxlX3NpemUlMkMlMjBtb2RlbC5jb25maWcuc2FtcGxlX3NpemUpJTBBbm9pc3lfc2FtcGxlLnNoYXBl",highlighted:`<span class="hljs-meta">>>> </span><span class="hljs-keyword">import</span> torch

	<span class="hljs-meta">>>> </span>torch.manual_seed(<span class="hljs-number">0</span>)

	<span class="hljs-meta">>>> </span>noisy_sample = torch.randn(<span class="hljs-number">1</span>, model.config.in_channels, model.config.sample_size, model.config.sample_size)
	<span class="hljs-meta">>>> </span>noisy_sample.shape
	torch.Size([<span class="hljs-number">1</span>, <span class="hljs-number">3</span>, <span class="hljs-number">256</span>, <span class="hljs-number">256</span>])`,lang:"py",wrap:!1});var _=s(U,4);a(_,{code:"d2l0aCUyMHRvcmNoLm5vX2dyYWQoKSUzQSUwQSUyMCUyMCUyMCUyMG5vaXN5X3Jlc2lkdWFsJTIwJTNEJTIwbW9kZWwoc2FtcGxlJTNEbm9pc3lfc2FtcGxlJTJDJTIwdGltZXN0ZXAlM0QyKS5zYW1wbGU=",highlighted:`<span class="hljs-meta">>>> </span><span class="hljs-keyword">with</span> torch.no_grad():
	<span class="hljs-meta">... </span> noisy_residual = model(sample=noisy_sample, timestep=<span class="hljs-number">2</span>).sample`,lang:"py",wrap:!1});var J=s(_,4);l(J,{title:"スケジューラ",local:"スケジューラ",headingTag:"h2"});var Z=s(J,8);a(Z,{code:"ZnJvbSUyMGRpZmZ1c2VycyUyMGltcG9ydCUyMEREUE1TY2hlZHVsZXIlMEElMEFzY2hlZHVsZXIlMjAlM0QlMjBERFBNU2NoZWR1bGVyLmZyb21fY29uZmlnKHJlcG9faWQpJTBBc2NoZWR1bGVy",highlighted:`<span class="hljs-meta">>>> </span><span class="hljs-keyword">from</span> diffusers <span class="hljs-keyword">import</span> DDPMScheduler

	<span class="hljs-meta">>>> </span>scheduler = DDPMScheduler.from_config(repo_id)
	<span class="hljs-meta">>>> </span>scheduler
	DDPMScheduler {
	<span class="hljs-string">"_class_name"</span>: <span class="hljs-string">"DDPMScheduler"</span>,
	<span class="hljs-string">"_diffusers_version"</span>: <span class="hljs-string">"0.13.1"</span>,
	<span class="hljs-string">"beta_end"</span>: <span class="hljs-number">0.02</span>,
	<span class="hljs-string">"beta_schedule"</span>: <span class="hljs-string">"linear"</span>,
	<span class="hljs-string">"beta_start"</span>: <span class="hljs-number">0.0001</span>,
	<span class="hljs-string">"clip_sample"</span>: true,
	<span class="hljs-string">"clip_sample_range"</span>: <span class="hljs-number">1.0</span>,
	<span class="hljs-string">"num_train_timesteps"</span>: <span class="hljs-number">1000</span>,
	<span class="hljs-string">"prediction_type"</span>: <span class="hljs-string">"epsilon"</span>,
	<span class="hljs-string">"trained_betas"</span>: null,
	<span class="hljs-string">"variance_type"</span>: <span class="hljs-string">"fixed_small"</span>
	}`,lang:"py",wrap:!1});var G=s(Z,10);a(G,{code:"bGVzc19ub2lzeV9zYW1wbGUlMjAlM0QlMjBzY2hlZHVsZXIuc3RlcChtb2RlbF9vdXRwdXQlM0Rub2lzeV9yZXNpZHVhbCUyQyUyMHRpbWVzdGVwJTNEMiUyQyUyMHNhbXBsZSUzRG5vaXN5X3NhbXBsZSkucHJldl9zYW1wbGUlMEFsZXNzX25vaXN5X3NhbXBsZS5zaGFwZQ==",highlighted:`<span class="hljs-meta">>>> </span>less_noisy_sample = scheduler.step(model_output=noisy_residual, timestep=<span class="hljs-number">2</span>, sample=noisy_sample).prev_sample
	<span class="hljs-meta">>>> </span>less_noisy_sample.shape`,lang:"py",wrap:!1});var T=s(G,8);a(T,{code:"aW1wb3J0JTIwUElMLkltYWdlJTBBaW1wb3J0JTIwbnVtcHklMjBhcyUyMG5wJTBBJTBBJTBBZGVmJTIwZGlzcGxheV9zYW1wbGUoc2FtcGxlJTJDJTIwaSklM0ElMEElMjAlMjAlMjAlMjBpbWFnZV9wcm9jZXNzZWQlMjAlM0QlMjBzYW1wbGUuY3B1KCkucGVybXV0ZSgwJTJDJTIwMiUyQyUyMDMlMkMlMjAxKSUwQSUyMCUyMCUyMCUyMGltYWdlX3Byb2Nlc3NlZCUyMCUzRCUyMChpbWFnZV9wcm9jZXNzZWQlMjAlMkIlMjAxLjApJTIwKiUyMDEyNy41JTBBJTIwJTIwJTIwJTIwaW1hZ2VfcHJvY2Vzc2VkJTIwJTNEJTIwaW1hZ2VfcHJvY2Vzc2VkLm51bXB5KCkuYXN0eXBlKG5wLnVpbnQ4KSUwQSUwQSUyMCUyMCUyMCUyMGltYWdlX3BpbCUyMCUzRCUyMFBJTC5JbWFnZS5mcm9tYXJyYXkoaW1hZ2VfcHJvY2Vzc2VkJTVCMCU1RCklMEElMjAlMjAlMjAlMjBkaXNwbGF5KGYlMjJJbWFnZSUyMGF0JTIwc3RlcCUyMCU3QmklN0QlMjIpJTBBJTIwJTIwJTIwJTIwZGlzcGxheShpbWFnZV9waWwp",highlighted:`<span class="hljs-meta">>>> </span><span class="hljs-keyword">import</span> PIL.Image
	<span class="hljs-meta">>>> </span><span class="hljs-keyword">import</span> numpy <span class="hljs-keyword">as</span> np


	<span class="hljs-meta">>>> </span><span class="hljs-keyword">def</span> <span class="hljs-title function_">display_sample</span>(<span class="hljs-params">sample, i</span>):
	<span class="hljs-meta">... </span> image_processed = sample.cpu().permute(<span class="hljs-number">0</span>, <span class="hljs-number">2</span>, <span class="hljs-number">3</span>, <span class="hljs-number">1</span>)
	<span class="hljs-meta">... </span> image_processed = (image_processed + <span class="hljs-number">1.0</span>) * <span class="hljs-number">127.5</span>
	<span class="hljs-meta">... </span> image_processed = image_processed.numpy().astype(np.uint8)

	<span class="hljs-meta">... </span> image_pil = PIL.Image.fromarray(image_processed[<span class="hljs-number">0</span>])
	<span class="hljs-meta">... </span> display(<span class="hljs-string">f"Image at step <span class="hljs-subst">{i}</span>"</span>)
	<span class="hljs-meta">... </span> display(image_pil)`,lang:"py",wrap:!1});var W=s(T,4);a(W,{code:"bW9kZWwudG8oJTIyY3VkYSUyMiklMEFub2lzeV9zYW1wbGUlMjAlM0QlMjBub2lzeV9zYW1wbGUudG8oJTIyY3VkYSUyMik=",highlighted:`<span class="hljs-meta">>>> </span>model.to(<span class="hljs-string">"cuda"</span>)
	<span class="hljs-meta">>>> </span>noisy_sample = noisy_sample.to(<span class="hljs-string">"cuda"</span>)`,lang:"py",wrap:!1});var k=s(W,4);a(k,{code:"aW1wb3J0JTIwdHFkbSUwQSUwQXNhbXBsZSUyMCUzRCUyMG5vaXN5X3NhbXBsZSUwQSUwQWZvciUyMGklMkMlMjB0JTIwaW4lMjBlbnVtZXJhdGUodHFkbS50cWRtKHNjaGVkdWxlci50aW1lc3RlcHMpKSUzQSUwQSUyMCUyMCUyMCUyMCUyMyUyMDEuJTIwcHJlZGljdCUyMG5vaXNlJTIwcmVzaWR1YWwlMEElMjAlMjAlMjAlMjB3aXRoJTIwdG9yY2gubm9fZ3JhZCgpJTNBJTBBJTIwJTIwJTIwJTIwJTIwJTIwJTIwJTIwcmVzaWR1YWwlMjAlM0QlMjBtb2RlbChzYW1wbGUlMkMlMjB0KS5zYW1wbGUlMEElMEElMjAlMjAlMjAlMjAlMjMlMjAyLiUyMGNvbXB1dGUlMjBsZXNzJTIwbm9pc3klMjBpbWFnZSUyMGFuZCUyMHNldCUyMHhfdCUyMC0lM0UlMjB4X3QtMSUwQSUyMCUyMCUyMCUyMHNhbXBsZSUyMCUzRCUyMHNjaGVkdWxlci5zdGVwKHJlc2lkdWFsJTJDJTIwdCUyQyUyMHNhbXBsZSkucHJldl9zYW1wbGUlMEElMEElMjAlMjAlMjAlMjAlMjMlMjAzLiUyMG9wdGlvbmFsbHklMjBsb29rJTIwYXQlMjBpbWFnZSUwQSUyMCUyMCUyMCUyMGlmJTIwKGklMjAlMkIlMjAxKSUyMCUyNSUyMDUwJTIwJTNEJTNEJTIwMCUzQSUwQSUyMCUyMCUyMCUyMCUyMCUyMCUyMCUyMGRpc3BsYXlfc2FtcGxlKHNhbXBsZSUyQyUyMGklMjAlMkIlMjAxKQ==",highlighted:`<span class="hljs-meta">>>> </span><span class="hljs-keyword">import</span> tqdm

	<span class="hljs-meta">>>> </span>sample = noisy_sample

	<span class="hljs-meta">>>> </span><span class="hljs-keyword">for</span> i, t <span class="hljs-keyword">in</span> <span class="hljs-built_in">enumerate</span>(tqdm.tqdm(scheduler.timesteps)):
	<span class="hljs-meta">... </span> <span class="hljs-comment"># 1. predict noise residual</span>
	<span class="hljs-meta">... </span> <span class="hljs-keyword">with</span> torch.no_grad():
	<span class="hljs-meta">... </span> residual = model(sample, t).sample

	<span class="hljs-meta">... </span> <span class="hljs-comment"># 2. compute less noisy image and set x_t -> x_t-1</span>
	<span class="hljs-meta">... </span> sample = scheduler.step(residual, t, sample).prev_sample

	<span class="hljs-meta">... </span> <span class="hljs-comment"># 3. optionally look at image</span>
	<span class="hljs-meta">... </span> <span class="hljs-keyword">if</span> (i + <span class="hljs-number">1</span>) % <span class="hljs-number">50</span> == <span class="hljs-number">0</span>:
	<span class="hljs-meta">... </span> display_sample(sample, i + <span class="hljs-number">1</span>)`,lang:"py",wrap:!1});var v=s(k,6);l(v,{title:"次のステップ",local:"次のステップ",headingTag:"h2"});var N=s(v,6);Y(N,{source:"https://github.com/huggingface/diffusers/blob/main/docs/source/ja/quicktour.md"}),A(2),S(C,e),F()}export{ls as component};

Xet Storage Details

Size:: 32.6 kB
Xet hash:: aa98a5781ec7e746c11ea19be467dd9ec22701e492d69f9c6a7a5f8c87afaa61

Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.