diff --git a/gigagan_pytorch/gigagan_pytorch.py b/gigagan_pytorch/gigagan_pytorch.py
index 5b2aa34..26d0544 100644
--- a/gigagan_pytorch/gigagan_pytorch.py
+++ b/gigagan_pytorch/gigagan_pytorch.py
@@ -1543,7 +1543,6 @@ def __init__(
 
         self.to_logits = nn.Sequential(
             conv2d_3x3(dim_last, dim_last),
-            leaky_relu(),
             Rearrange('b c h w -> b (c h w)'),
             nn.Linear(dim_last * (4 ** 2), 1),
             Rearrange('b 1 -> b')
diff --git a/gigagan_pytorch/version.py b/gigagan_pytorch/version.py
index a2570eb..dce113e 100644
--- a/gigagan_pytorch/version.py
+++ b/gigagan_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.2.15'
+__version__ = '0.2.16'