mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-15 16:02:32 +08:00
Fix.
This commit is contained in:
parent
1840bc5226
commit
bb0655140f
@ -234,7 +234,7 @@ class Encoder(nn.Module):
|
|||||||
if self.refiner_vae:
|
if self.refiner_vae:
|
||||||
xl = [x[:, :, :1, :, :]]
|
xl = [x[:, :, :1, :, :]]
|
||||||
if x.shape[2] > self.ffactor_temporal:
|
if x.shape[2] > self.ffactor_temporal:
|
||||||
xl += torch.split(x[:, :, 1: 1 + ((x.shape[2] - 1) // self.ffactor_temporal) * self.ffactor_temporal, :, :], self.ffactor_temporal, dim=2)
|
xl += torch.split(x[:, :, 1: 1 + ((x.shape[2] - 1) // self.ffactor_temporal) * self.ffactor_temporal, :, :], self.ffactor_temporal * 2, dim=2)
|
||||||
x = xl
|
x = xl
|
||||||
else:
|
else:
|
||||||
x = [x]
|
x = [x]
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user