298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-16 16:43:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Uni pc sampler now works with audio and video models.

Browse Source
This commit is contained in:
comfyanonymous 2025-01-18 05:27:58 -05:00
parent 2f3ab40b62
commit 507199d9a8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy/extra_samplers/uni_pc.py
View File

@ -661,7 +661,7 @@ class UniPC:
if x_t is None: if x_t is None:
if use_predictor: if use_predictor:
pred_res = torch.einsum('k,bkchw->bchw', rhos_p, D1s) pred_res = torch.tensordot(D1s, rhos_p, dims=([1], [0])) # torch.einsum('k,bkchw->bchw', rhos_p, D1s)
else: else:
pred_res = 0 pred_res = 0
x_t = x_t_ - expand_dims(alpha_t * B_h, dims) * pred_res x_t = x_t_ - expand_dims(alpha_t * B_h, dims) * pred_res
@ -669,7 +669,7 @@ class UniPC:
if use_corrector: if use_corrector:
model_t = self.model_fn(x_t, t) model_t = self.model_fn(x_t, t)
if D1s is not None: if D1s is not None:
corr_res = torch.einsum('k,bkchw->bchw', rhos_c[:-1], D1s) corr_res = torch.tensordot(D1s, rhos_c[:-1], dims=([1], [0])) # torch.einsum('k,bkchw->bchw', rhos_c[:-1], D1s)
else: else:
corr_res = 0 corr_res = 0
D1_t = (model_t - model_prev_0) D1_t = (model_t - model_prev_0)