Buckets:

hf-doc-build/doc / diffusers /main /ru /modular_diffusers /modular_diffusers_states.md
mishig's picture
|
download
raw
5.3 kB
# Состояния
Блоки полагаются на структуры данных [PipelineState](/docs/diffusers/main/ru/api/modular_diffusers/pipeline_states#diffusers.modular_pipelines.PipelineState) и [BlockState](/docs/diffusers/main/ru/api/modular_diffusers/pipeline_states#diffusers.modular_pipelines.BlockState) для обмена и общего использования данных.
| Состояние | Описание |
|-------|-------------|
| [PipelineState](/docs/diffusers/main/ru/api/modular_diffusers/pipeline_states#diffusers.modular_pipelines.PipelineState) | Хранит общие данные, необходимые для выполнения конвейера, и позволяет блокам читать и обновлять эти данные. |
| [BlockState](/docs/diffusers/main/ru/api/modular_diffusers/pipeline_states#diffusers.modular_pipelines.BlockState) | Позволяет каждому блоку выполнять свои вычисления с необходимыми данными из `inputs`|
В этом руководстве объясняется, как работают состояния и как они связывают блоки.
## PipelineState
[PipelineState](/docs/diffusers/main/ru/api/modular_diffusers/pipeline_states#diffusers.modular_pipelines.PipelineState) — это глобальный контейнер состояния для всех блоков. Он хранит полное состояние конвейера во время выполнения и предоставляет структурированный способ для блоков читать и записывать общие данные.
[PipelineState](/docs/diffusers/main/ru/api/modular_diffusers/pipeline_states#diffusers.modular_pipelines.PipelineState) хранит все данные в словаре `values`, который представляет собой **изменяемое** состояние, содержащее входные значения, предоставленные пользователем, и промежуточные выходные значения, генерируемые блоками. Если блок изменяет `input`, это изменение отразится в словаре `values` после вызова `set_block_state`.
```py
PipelineState(
values={
'prompt': 'a cat'
'guidance_scale': 7.0
'num_inference_steps': 25
'prompt_embeds': Tensor(dtype=torch.float32, shape=torch.Size([1, 1, 1, 1]))
'negative_prompt_embeds': None
},
)
```
## BlockState
[BlockState](/docs/diffusers/main/ru/api/modular_diffusers/pipeline_states#diffusers.modular_pipelines.BlockState) представляет собой локальный вид тех переменных, которые отдельному блоку необходимы из [PipelineState](/docs/diffusers/main/ru/api/modular_diffusers/pipeline_states#diffusers.modular_pipelines.PipelineState) для выполнения вычислений.
Обращайтесь к этим переменным напрямую как к атрибутам, например `block_state.image`.
```py
BlockState(
image: <PIL.Image.Image image mode=RGB size=512x512 at 0x7F3ECC494640>
)
```
Когда выполняется метод `__call__` блока, он получает `BlockState` с помощью `self.get_block_state(state)`, выполняет свои операции и обновляет [PipelineState](/docs/diffusers/main/ru/api/modular_diffusers/pipeline_states#diffusers.modular_pipelines.PipelineState) с помощью `self.set_block_state(state, block_state)`.
```py
def __call__(self, components, state):
# retrieve BlockState
block_state = self.get_block_state(state)
# computation logic on inputs
# update PipelineState
self.set_block_state(state, block_state)
return components, state
```
## Взаимодействие состояния
Взаимодействие [PipelineState](/docs/diffusers/main/ru/api/modular_diffusers/pipeline_states#diffusers.modular_pipelines.PipelineState) и [BlockState](/docs/diffusers/main/ru/api/modular_diffusers/pipeline_states#diffusers.modular_pipelines.BlockState) определяется параметрами `inputs` и `intermediate_outputs` блока.
- `inputs`, блок может изменить входной параметр — например, `block_state.image` — и это изменение может быть глобально передано в [PipelineState](/docs/diffusers/main/ru/api/modular_diffusers/pipeline_states#diffusers.modular_pipelines.PipelineState) с помощью вызова `set_block_state`.
- `intermediate_outputs` — это новая переменная, создаваемая блоком. Она добавляется в словарь `values` объекта [PipelineState](/docs/diffusers/main/ru/api/modular_diffusers/pipeline_states#diffusers.modular_pipelines.PipelineState) и доступна для последующих блоков или может быть получена пользователем в качестве итогового вывода конвейера.

Xet Storage Details

Size:
5.3 kB
·
Xet hash:
eed47b253f4dcdb94850d13ef198de04d07eb0a2520aaf1c7f41cf5e6f59864a

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