"tests/vscode:/vscode.git/clone" did not exist on "c73e609aae8052c89d5b3e4c26be3a6c68481ecd"
Commit 79040635 authored by comfyanonymous's avatar comfyanonymous
Browse files

Remove unnecessary code.

parent 66d35c07
...@@ -809,7 +809,7 @@ class UNetModel(nn.Module): ...@@ -809,7 +809,7 @@ class UNetModel(nn.Module):
self.out = nn.Sequential( self.out = nn.Sequential(
operations.GroupNorm(32, ch, dtype=self.dtype, device=device), operations.GroupNorm(32, ch, dtype=self.dtype, device=device),
nn.SiLU(), nn.SiLU(),
zero_module(operations.conv_nd(dims, model_channels, out_channels, 3, padding=1, dtype=self.dtype, device=device)), operations.conv_nd(dims, model_channels, out_channels, 3, padding=1, dtype=self.dtype, device=device),
) )
if self.predict_codebook_ids: if self.predict_codebook_ids:
self.id_predictor = nn.Sequential( self.id_predictor = nn.Sequential(
......
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