# HeliosDMDScheduler

`HeliosDMDScheduler` được xây dựng dựa trên phương pháp lấy mẫu khớp dòng hình chóp (pyramidal flow-matching sampling) được giới thiệu trong [Helios](https://huggingface.co/papers).

## HeliosDMDScheduler[[diffusers.HeliosDMDScheduler]]

#### diffusers.HeliosDMDScheduler[[diffusers.HeliosDMDScheduler]]

```python
diffusers.HeliosDMDScheduler(num_train_timesteps: int = 1000, shift: float = 1.0, stages: int = 3, stage_range: list = [0, 0.3333333333333333, 0.6666666666666666, 1], gamma: float = 0.3333333333333333, prediction_type: str = 'flow_prediction', use_flow_sigmas: bool = True, use_dynamic_shifting: bool = False, time_shift_type: typing.Literal['exponential', 'linear'] = 'linear')
```

[Nguồn](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/schedulers/scheduling_helios_dmd.py#L35)

#### init_sigmas[[diffusers.HeliosDMDScheduler.init_sigmas]]

```python
init_sigmas()
```

[Nguồn](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/schedulers/scheduling_helios_dmd.py#L69)

khởi tạo các bước thời gian và sigma toàn cục

#### init_sigmas_for_each_stage[[diffusers.HeliosDMDScheduler.init_sigmas_for_each_stage]]

```python
init_sigmas_for_each_stage()
```

[Nguồn](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/schedulers/scheduling_helios_dmd.py#L87)

Khởi tạo các bước thời gian cho từng giai đoạn

#### set_begin_index[[diffusers.HeliosDMDScheduler.set_begin_index]]

```python
set_begin_index(begin_index: int = 0)
```

[Nguồn](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/schedulers/scheduling_helios_dmd.py#L161)

**Tham số:**

begin_index (`int`) : Chỉ số bắt đầu cho trình lập lịch.

Đặt chỉ số bắt đầu cho trình lập lịch. Hàm này nên được chạy từ pipeline trước khi suy luận.

#### set_timesteps[[diffusers.HeliosDMDScheduler.set_timesteps]]

```python
set_timesteps(num_inference_steps: int, stage_index: int | None = None, device: typing.Union[str, torch.device] = None, sigmas: bool | None = None, mu: bool | None = None, is_amplify_first_chunk: bool = False)
```

[Nguồn](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/schedulers/scheduling_helios_dmd.py#L174)

Đặt các bước thời gian và sigma cho từng giai đoạn

#### time_shift[[diffusers.HeliosDMDScheduler.time_shift]]

```python
time_shift(mu: float, sigma: float, t: Tensor)
```

[Nguồn](https://github.com/huggingface/diffusers/blob/v0.43.0/src/diffusers/schedulers/scheduling_helios_dmd.py#L236)

**Tham số:**

mu (`float`) : Tham số mu cho việc dịch chuyển thời gian.

sigma (`float`) : Tham số sigma cho phép dịch chuyển thời gian.

t (`torch.Tensor`) : Các bước thời gian đầu vào.

**Trả về:** `torch.Tensor`

Các bước thời gian đã dịch chuyển.

Áp dụng dịch chuyển thời gian vào các sigma.

scheduling_helios_dmd