diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-03-12 08:24:17 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-12 08:24:17 +0300 |
| commit | beb96bd11534aaafd6ebc6c287013a4c3e1d5a95 (patch) | |
| tree | 858d034c5c51fc965ece97159e3717644b60ca79 /modules/models/diffusion | |
| parent | bbc4b0478ab24e69c94060d81ef778fcbe087b57 (diff) | |
| parent | 48f4abd2e61e545104f72eb50c9ab9b100726948 (diff) | |
Merge pull request #8515 from EllangoK/unipc-typo
Fix dims typo in unipc
Diffstat (limited to 'modules/models/diffusion')
| -rw-r--r-- | modules/models/diffusion/uni_pc/uni_pc.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/models/diffusion/uni_pc/uni_pc.py b/modules/models/diffusion/uni_pc/uni_pc.py index df63d1bc..e9a093a2 100644 --- a/modules/models/diffusion/uni_pc/uni_pc.py +++ b/modules/models/diffusion/uni_pc/uni_pc.py @@ -719,7 +719,7 @@ class UniPC: x_t = x_t_ - expand_dims(alpha_t * B_h, dims) * (corr_res + rhos_c[-1] * D1_t) else: x_t_ = ( - expand_dims(torch.exp(log_alpha_t - log_alpha_prev_0), dimss) * x + expand_dims(torch.exp(log_alpha_t - log_alpha_prev_0), dims) * x - expand_dims(sigma_t * h_phi_1, dims) * model_prev_0 ) if x_t is None: |