mirror of
https://github.com/facefusion/facefusion-labs.git
synced 2026-04-19 15:56:37 +02:00
Fix test
This commit is contained in:
@@ -65,7 +65,7 @@ class AAD(nn.Module):
|
||||
temp_tensor = layer(temp_tensors, source_embedding, target_attribute)
|
||||
temp_tensors = nn.functional.interpolate(temp_tensor, scale_factor = 2, mode = 'bilinear', align_corners = False)
|
||||
|
||||
target_attribute = target_attributes[-1],
|
||||
target_attribute = target_attributes[-1]
|
||||
temp_tensors = self.layers[-1](temp_tensors, source_embedding, target_attribute)
|
||||
output_tensor = torch.tanh(temp_tensors)
|
||||
return output_tensor
|
||||
|
||||
Reference in New Issue
Block a user