From 71e0ae34c03cbab17fa5e24b615ebd5830da27fc Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Wed, 12 Feb 2025 15:46:33 +0100
Subject: [PATCH] Fix self

---
 face_swapper/src/models/discriminator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/face_swapper/src/models/discriminator.py b/face_swapper/src/models/discriminator.py
index 63da839..5a2865c 100644
--- a/face_swapper/src/models/discriminator.py
+++ b/face_swapper/src/models/discriminator.py
@@ -44,7 +44,7 @@ class MultiscaleDiscriminator(nn.Module):
 class NLayerDiscriminator(nn.Module):
 	def __init__(self, input_channels : int, num_filters : int, num_layers : int, kernel_size : int) -> None:
 		super(NLayerDiscriminator, self).__init__()
-		layers = self.create_layers(input_channels, num_filters, num_layers, kernel_size)
+		layers = self.create_layers(self, input_channels, num_filters, num_layers, kernel_size)
 		self.discriminator = nn.Sequential(*layers)
 
 	@staticmethod