mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-08-31 00:10:39 +02:00
fix
This commit is contained in:
@@ -46,7 +46,7 @@ class MaskNet(nn.Module):
|
||||
output_tensor = up_sample(output_tensor)
|
||||
|
||||
output_tensor = self.conv(output_tensor)
|
||||
output_tensor = self.activation(output_tensor)
|
||||
output_tensor = self.sigmoid(output_tensor)
|
||||
return output_tensor
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user