diff --git a/.gitignore b/.gitignore
index beb7476..8b14032 100644
--- a/.gitignore
+++ b/.gitignore
@@ -20,3 +20,6 @@ samples/
 
 # Inference outputs
 output/
+
+# OS files
+.DS_Store
diff --git a/scripts/dump_pytorch_reference.py b/scripts/dump_pytorch_reference.py
index f91b70a..8fbbc25 100644
--- a/scripts/dump_pytorch_reference.py
+++ b/scripts/dump_pytorch_reference.py
@@ -158,13 +158,13 @@ class CNNRefinerModule(nn.Module):
 class GreenFormer(nn.Module):
     """Top-level CorridorKey model: Hiera encoder + dual decoder heads + CNN refiner."""
 
-    def __init__(self) -> None:
+    def __init__(self, img_size: int = IMG_SIZE) -> None:
         super().__init__()
         self.encoder = timm.create_model(
             HIERA_MODEL_NAME,
             pretrained=False,
             features_only=True,
-            img_size=IMG_SIZE,
+            img_size=img_size,
         )
         self._patch_first_conv()
 
diff --git a/scripts/infer_pytorch.py b/scripts/infer_pytorch.py
index a126364..b50b5a7 100644
--- a/scripts/infer_pytorch.py
+++ b/scripts/infer_pytorch.py
@@ -63,7 +63,7 @@ def main() -> None:
     torch.set_grad_enabled(False)
 
     print(f"Building model (img_size={args.img_size})...")
-    model = GreenFormer()
+    model = GreenFormer(img_size=args.img_size)
     model.eval()
 
     print(f"Loading checkpoint: {args.checkpoint}")