mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-05-22 23:59:40 +02:00
fix
This commit is contained in:
@@ -14,6 +14,7 @@ split_ratio =
|
||||
embedder_path =
|
||||
gazer_path =
|
||||
motion_extractor_path =
|
||||
parser_path =
|
||||
|
||||
[training.model.generator]
|
||||
identity_channels =
|
||||
|
||||
@@ -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