Buckets:
Pipeline 块
ModularPipelineBlocks[[diffusers.ModularPipelineBlocks]]
diffusers.ModularPipelineBlocks[[diffusers.ModularPipelineBlocks]]
diffusers.ModularPipelineBlocks()
所有 Pipeline Blocks 的基类:ConditionalPipelineBlocks、AutoPipelineBlocks、SequentialPipelineBlocks、LoopSequentialPipelineBlocks
ModularPipelineBlocks 提供了用于加载和保存流水线块定义的方法。
get_block_state[[diffusers.ModularPipelineBlocks.get_block_state]]
get_block_state(state: PipelineState)
在一个字典中获取所有输入和中间值
get_execution_blocks[[diffusers.ModularPipelineBlocks.get_execution_blocks]]
get_execution_blocks(**kwargs)
参数:
- **kwargs:输入名称和值。只有触发输入会影响块的选择。
获取在给定输入下将执行的块。支持条件块选择的子类必须实现此方法。
get_workflow[[diffusers.ModularPipelineBlocks.get_workflow]]
get_workflow(workflow_name: str)
参数:
workflow_name : 要获取的工作流名称。
获取特定工作流的执行块。必须由定义了 _workflow_map 的子类实现。
init_pipeline[[diffusers.ModularPipelineBlocks.init_pipeline]]
init_pipeline(pretrained_model_name_or_path: str | os.PathLike | None = None, components_manager: diffusers.modular_pipelines.components_manager.ComponentsManager | None = None, collection: str | None = None)
创建一个 ModularPipeline,可选地接受 pretrained_model_name_or_path 以从 hub 加载。
SequentialPipelineBlocks[[diffusers.SequentialPipelineBlocks]]
diffusers.SequentialPipelineBlocks[[diffusers.SequentialPipelineBlocks]]
diffusers.SequentialPipelineBlocks()
参数:
block_classes : 要使用的块类列表
block_names : 每个块的 prefixes 列表
A Pipeline Blocks 将多个 pipeline block 类组合为一个。调用时,它会按顺序调用每个 block。
此类继承自 ModularPipelineBlocks。请查阅父类文档,了解该库为所有 pipeline 块实现的通用方法(例如加载或保存等)。
from_blocks_dict[[diffusers.SequentialPipelineBlocks.from_blocks_dict]]
from_blocks_dict(blocks_dict: dict[str, typing.Any], description: str | None = None)
参数:
blocks_dict : 将块名称映射到块类或实例的字典
返回值:
新的 SequentialPipelineBlocks 实例
从块字典创建 SequentialPipelineBlocks 实例。
get_execution_blocks[[diffusers.SequentialPipelineBlocks.get_execution_blocks]]
get_execution_blocks(**kwargs)
参数:
- **kwargs:输入名称和值。只有触发输入会影响块的选择。
返回值:
仅包含将执行的块的 SequentialPipelineBlocks
获取在给定指定输入的情况下将执行的块。
当遍历按顺序处理各个块时,已解析块的中间输出会被添加到活动输入中。这意味着,只要条件是基于存在性(即 None 或非 None)而非实际值,依赖于中间结果的有条件块(例如“如果 image_latents 存在,则运行 img2img”)就能正确解析。
LoopSequentialPipelineBlocks[[diffusers.LoopSequentialPipelineBlocks]]
diffusers.LoopSequentialPipelineBlocks[[diffusers.LoopSequentialPipelineBlocks]]
diffusers.LoopSequentialPipelineBlocks()
参数:
block_classes : 要使用的块类列表
block_names : 每个块的 prefixes 列表
Pipeline 是一种将多个 pipeline 块类组合到 For Loop 中的块。调用时,它会按顺序调用每个块。
此类继承自 ModularPipelineBlocks。请查阅父类文档,了解该库为所有 pipeline 块实现的通用方法(例如加载或保存等)。
from_blocks_dict[[diffusers.LoopSequentialPipelineBlocks.from_blocks_dict]]
from_blocks_dict(blocks_dict: dict[str, typing.Any])
参数:
blocks_dict : 将块名称映射到块实例的字典
返回值:
新的 LoopSequentialPipelineBlocks 实例
从块字典创建 LoopSequentialPipelineBlocks 实例。
AutoPipelineBlocks[[diffusers.AutoPipelineBlocks]]
diffusers.AutoPipelineBlocks[[diffusers.AutoPipelineBlocks]]
diffusers.AutoPipelineBlocks()
参数:
block_classes : 要使用的块类列表。其长度必须与 block_names 和 block_trigger_inputs 相同。
block_names : 每个块的名称列表。其长度必须与 block_classes 和 block_trigger_inputs 相同。
block_trigger_inputs : 输入名称列表,其中每个元素指定对应块的触发输入。使用 None 标记默认块。
Pipeline 块会根据触发输入的存在情况,自动选择要运行的块。
这是 ConditionalPipelineBlocks 的一个专用版本,其中:
- 每个块都有一个对应的触发输入(1:1 映射)
- 块选择是自动的:第一个触发输入存在的块会被选中
block_trigger_inputs的长度必须与block_names和block_classes相同- 在
block_trigger_inputs中使用None来指定默认块,即在没有触发输入时将运行的块
示例:
class MyAutoBlock(AutoPipelineBlocks):
block_classes = [InpaintEncoderBlock, ImageEncoderBlock, TextEncoderBlock]
block_names = ["inpaint", "img2img", "text2img"]
block_trigger_inputs = ["mask_image", "image", None] # text2img is the default
使用此定义:
- 只要提供了
mask_image,"inpaint" 块就会执行(无论是否提供了image)。 - 如果未提供
mask_image但提供了image,则会运行 "img2img" 块 - 否则,将执行 "text2img" 块(默认值,触发器为
None)
select_block[[diffusers.AutoPipelineBlocks.select_block]]
select_block(**kwargs)
根据哪个触发输入存在(不为 None)来选择块。
ConditionalPipelineBlocks[[diffusers.ConditionalPipelineBlocks]]
diffusers.ConditionalPipelineBlocks[[diffusers.ConditionalPipelineBlocks]]
diffusers.ConditionalPipelineBlocks()
参数:
block_classes : 要使用的块类列表。其长度必须与 block_names 相同。
block_names : 每个块的名称列表。其长度必须与 block_classes 相同。
block_trigger_inputs : select_block() 用于确定要运行哪个块的输入名称列表。对于 ConditionalPipelineBlocks,此列表无需与 block_names 和 block_classes 对应。对于 AutoPipelineBlocks,此列表的长度必须与 block_names 和 block_classes 相同,其中每个元素指定对应块的触发输入。
default_block_name : 当没有触发输入匹配时,要运行的默认块的名称。如果为 None,则在未提供触发输入时,可以完全跳过此块。
Pipeline Blocks 会根据输入有条件地选择要运行的 block。子类必须实现 select_block 方法以定义选择 block 的逻辑。目前,我们仅支持基于输入是否存在(即是否为 None)的选择逻辑。
此类继承自 ModularPipelineBlocks。请查阅父类文档,了解该库为所有 pipeline 块实现的通用方法(例如加载或保存等)。
get_execution_blocks[[diffusers.ConditionalPipelineBlocks.get_execution_blocks]]
get_execution_blocks(**kwargs)
参数:
- **kwargs:输入名称和值。只有触发输入会影响块的选择。
返回: - ModularPipelineBlocks
叶块或已解析的 SequentialPipelineBlocks
None:如果此块将被跳过(没有匹配的触发器且没有默认值)
获取给定输入下将执行的块。
递归解析嵌套的 ConditionalPipelineBlocks,直到达到以下任一条件:
- 叶块(没有 sub_blocks 或 LoopSequentialPipelineBlocks)→ 返回单个
ModularPipelineBlocks SequentialPipelineBlocks→ 委托给其get_execution_blocks(),该方法返回 一个包含已解析执行块的SequentialPipelineBlocks
select_block[[diffusers.ConditionalPipelineBlocks.select_block]]
select_block(**kwargs)
参数:
- **kwargs : 来自 state 的触发器输入名称及其值。
返回: str | None
要运行的块名称,或 None 以使用默认值/跳过。
根据触发器输入选择要运行的块。子类必须实现此方法以定义选择块的逻辑。
注意:当触发输入包含来自前面块的中间输出时,选择逻辑应仅取决于该输入是否存在(即是否为 None),而不取决于其实际值。这是因为 get_execution_blocks() 通过传播中间输出名称(而不传播其运行时值)来静态解析条件。
Xet Storage Details
- Size:
- 10.9 kB
- Xet hash:
- 4010c1f6632feadf715ab3c3b229a3120b582403ceaaa21a4ea7002e4dc8b839
Xet efficiently stores files, intelligently splitting them into unique chunks and accelerating uploads and downloads. More info.