"...resnet50_tensorflow.git" did not exist on "9ac54873f7ac15cec6e203a30b72615c3dfc1d75"
Commit c6e34963 authored by comfyanonymous's avatar comfyanonymous
Browse files

Make t2i adapter work with any latent resolution.

parent a8705dbf
...@@ -56,7 +56,12 @@ class Downsample(nn.Module): ...@@ -56,7 +56,12 @@ class Downsample(nn.Module):
def forward(self, x): def forward(self, x):
assert x.shape[1] == self.channels assert x.shape[1] == self.channels
return self.op(x) if not self.use_conv:
padding = [x.shape[2] % 2, x.shape[3] % 2]
self.op.padding = padding
x = self.op(x)
return x
class ResnetBlock(nn.Module): class ResnetBlock(nn.Module):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment