mirror of
https://git.datalinker.icu/comfyanonymous/ComfyUI
synced 2026-09-02 22:27:18 +08:00
'save'
This commit is contained in:
parent
5487a54578
commit
e76c8ed68b
@ -1157,7 +1157,8 @@ def patch_motion(
|
|||||||
print("point_feature shape:", point_feature.shape)
|
print("point_feature shape:", point_feature.shape)
|
||||||
out_feature = merge_final(point_feature, vert_weight, vert_index).permute(3, 0, 1, 2) # T - 1, H, W, C => C, T - 1, H, W
|
out_feature = merge_final(point_feature, vert_weight, vert_index).permute(3, 0, 1, 2) # T - 1, H, W, C => C, T - 1, H, W
|
||||||
out_weight = vert_weight.sum(-1) # T - 1, H, W
|
out_weight = vert_weight.sum(-1) # T - 1, H, W
|
||||||
|
|
||||||
|
print("out_weight shape:", out_weight)
|
||||||
# out feature -> already soft weighted
|
# out feature -> already soft weighted
|
||||||
mix_feature = out_feature + vid[vae_divide[0]:, 1:] * (1 - out_weight.clamp(0, 1))
|
mix_feature = out_feature + vid[vae_divide[0]:, 1:] * (1 - out_weight.clamp(0, 1))
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user