diff options
| author | EllangoK <karun.ellango7@gmail.com> | 2023-03-11 15:52:14 -0500 |
|---|---|---|
| committer | EllangoK <karun.ellango7@gmail.com> | 2023-03-11 15:52:14 -0500 |
| commit | 48f4abd2e61e545104f72eb50c9ab9b100726948 (patch) | |
| tree | bd684c5b3cc146a5d6505ef62d6f651d585d5ce9 /modules/models/diffusion/uni_pc | |
| parent | 27e319dc4f09a2f040043948e5c52965976f8491 (diff) | |
fix dims typo in unipc
Diffstat (limited to 'modules/models/diffusion/uni_pc')
| -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: |