Skip to content

Commit 88badfa

Browse files
committed
fix
1 parent f87d065 commit 88badfa

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

modules/conform/unet_with_conform.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -620,8 +620,8 @@ def forward(self, x, pitch=None, mask=None):
620620
midiout = self.outln(xo)
621621
cutprp = torch.sigmoid(cutprp)
622622
cutprp = torch.squeeze(cutprp, -1)
623-
if self.sig:
624-
midiout = torch.sigmoid(midiout)
623+
# if self.sig:
624+
# midiout = torch.sigmoid(midiout)
625625
return midiout, cutprp
626626

627627

0 commit comments

Comments
 (0)