Buckets:

mishig's picture
|
download
raw
5.52 kB
# AutoPipeline
[AutoPipeline](../api/pipelines/auto_pipeline) は、タスクに基づいて適切なパイプラインのサブクラスを自動的に選択する、*タスクとモデル*を扱うパイプラインです。具体的なパイプラインのサブクラス名を知らなくても、異なるパイプラインのサブクラスを読み込む複雑さを処理します。
これは、モデルに基づいてパイプラインのサブクラスを自動的に選択する*モデル専用*のパイプラインである[DiffusionPipeline](/docs/diffusers/v0.43.0/ja/api/pipelines/overview#diffusers.DiffusionPipeline)とは異なります。
```text
AutoPipelineForImage2Image.from_pretrained(model_id)
|
+-- read model_index.json (e.g. StableDiffusionXLPipeline)
+-- task mapping (image-to-image)
|
v
StableDiffusionXLImg2ImgPipeline // returned instance
```
[AutoPipelineForImage2Image](/docs/diffusers/v0.43.0/ja/api/pipelines/auto_pipeline#diffusers.AutoPipelineForImage2Image) は、タスク固有のサブクラス(例: [StableDiffusionXLImg2ImgPipeline](/docs/diffusers/v0.43.0/ja/api/pipelines/stable_diffusion/stable_diffusion_xl#diffusers.StableDiffusionXLImg2ImgPipeline))を返します。このサブクラスは、画像から画像へのタスクでのみ使用できます。
```py
import torch
from diffusers import AutoPipelineForImage2Image
pipeline = AutoPipelineForImage2Image.from_pretrained(
"RunDiffusion/Juggernaut-XL-v9", dtype=torch.bfloat16, device_map="cuda", # or "mps", "xpu", "cpu"
)
print(pipeline)
# StableDiffusionXLImg2ImgPipeline {
# "_class_name": "StableDiffusionXLImg2ImgPipeline",
# ...
# }
```
[DiffusionPipeline](/docs/diffusers/v0.43.0/ja/api/pipelines/overview#diffusers.DiffusionPipeline)で同じモデルを読み込むと、デフォルトのテキストから画像へのサブクラスである[StableDiffusionXLPipeline](/docs/diffusers/v0.43.0/ja/api/pipelines/stable_diffusion/stable_diffusion_xl#diffusers.StableDiffusionXLPipeline)が返されます。このパイプラインはテキストから画像への生成用です。画像から画像への生成やインペインティングを行う場合は、[AutoPipelineForImage2Image](/docs/diffusers/v0.43.0/ja/api/pipelines/auto_pipeline#diffusers.AutoPipelineForImage2Image)や[AutoPipelineForInpainting](/docs/diffusers/v0.43.0/ja/api/pipelines/auto_pipeline#diffusers.AutoPipelineForInpainting)などのタスク用AutoPipeline、または対応するタスク固有のサブクラスを読み込んでください。
```py
import torch
from diffusers import DiffusionPipeline
pipeline = DiffusionPipeline.from_pretrained(
"RunDiffusion/Juggernaut-XL-v9", dtype=torch.bfloat16, device_map="cuda", # or "mps", "xpu", "cpu"
)
print(pipeline)
# StableDiffusionXLPipeline {
# "_class_name": "StableDiffusionXLPipeline",
# ...
# }
```
## from_pipe でタスクを切り替える
タスクのAutoPipelineを一度読み込み、[from_pipe()](/docs/diffusers/v0.43.0/ja/api/pipelines/auto_pipeline#diffusers.AutoPipelineForImage2Image.from_pipe)でタスクを切り替えることで、重みの再ダウンロードなしに切り替えが可能です。コンポーネントはソースのパイプラインから再利用されます。
```py
import torch
from diffusers import AutoPipelineForText2Image, AutoPipelineForImage2Image
pipeline_t2i = AutoPipelineForText2Image.from_pretrained(
"RunDiffusion/Juggernaut-XL-v9", dtype=torch.bfloat16, device_map="cuda", # or "mps", "xpu", "cpu"
)
pipeline_i2i = AutoPipelineForImage2Image.from_pipe(pipeline_t2i)
```
詳細については、[複数のパイプラインでのモデルの再利用](../using-diffusers/loading#reusing-models-in-multiple-pipelines)を参照してください。
モデルがサポートされているかどうかを確認するには、[mappings](https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/auto_pipeline.py)を参照してください。サポートされていないモデルの読み込みに失敗すると、エラーが返されます。
```py
import torch
from diffusers import AutoPipelineForImage2Image
pipeline = AutoPipelineForImage2Image.from_pretrained(
"openai/shap-e-img2img", dtype=torch.float16,
)
"ValueError: AutoPipeline can't find a pipeline linked to ShapEImg2ImgPipeline for None"
```
[AutoPipeline](../api/pipelines/auto_pipeline) クラスには4つの種類があります。
- [AutoPipelineForText2Image](/docs/diffusers/v0.43.0/ja/api/pipelines/auto_pipeline#diffusers.AutoPipelineForText2Image)
- [AutoPipelineForImage2Image](/docs/diffusers/v0.43.0/ja/api/pipelines/auto_pipeline#diffusers.AutoPipelineForImage2Image)
- [AutoPipelineForInpainting](/docs/diffusers/v0.43.0/ja/api/pipelines/auto_pipeline#diffusers.AutoPipelineForInpainting)
- [AutoPipelineForText2Audio](/docs/diffusers/v0.43.0/ja/api/pipelines/auto_pipeline#diffusers.AutoPipelineForText2Audio)
これらのクラスにはそれぞれ事前定義されたマッピングがあり、パイプラインをタスク固有のサブクラスにリンクしています。
[from_pretrained()](/docs/diffusers/v0.43.0/ja/api/pipelines/auto_pipeline#diffusers.AutoPipelineForText2Image.from_pretrained) が呼び出されると、`model_index.json` ファイルからクラス名を抽出し、マッピングに基づいてタスクに適した pipeline のサブクラスを選択します。

Xet Storage Details

Size:
5.52 kB
·
Xet hash:
923f75938177f9b1c5e61ebb24aa812df344a1a6bbd6f4457a04412169836d38

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