"doc/vscode:/vscode.git/clone" did not exist on "5df1f7e2e568c37a27a936720f4fd9f51e00a318"
Commit 9d54066e authored by comfyanonymous's avatar comfyanonymous
Browse files

This isn't needed for inference.

parent fa2cca05
......@@ -610,13 +610,12 @@ class SpatialTransformer(nn.Module):
for d in range(depth)]
)
if not use_linear:
self.proj_out = zero_module(nn.Conv2d(inner_dim,
in_channels,
self.proj_out = nn.Conv2d(inner_dim,in_channels,
kernel_size=1,
stride=1,
padding=0))
padding=0)
else:
self.proj_out = zero_module(comfy.ops.Linear(in_channels, inner_dim))
self.proj_out = comfy.ops.Linear(in_channels, inner_dim)
self.use_linear = use_linear
def forward(self, x, context=None, transformer_options={}):
......
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