From 6129756e106e73ce6af656b8875c4a7ea1001ed9 Mon Sep 17 00:00:00 2001
From: Luciferian Ink <LuciferianInk@protonmail.com>
Date: Fri, 17 Jan 2025 18:04:25 -0600
Subject: [PATCH] remove unused argument

---
 bytelatent/model/local_models.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/bytelatent/model/local_models.py b/bytelatent/model/local_models.py
index 59fa76d..c16f62e 100644
--- a/bytelatent/model/local_models.py
+++ b/bytelatent/model/local_models.py
@@ -199,9 +199,6 @@ class LocalModelBase(nn.Module):
 class LocalEncoder(LocalModelBase):
     def __init__(self, args: LocalModelArgs):
         super().__init__(args)
-        self.output_proj = (
-            args.patching_mode in ["entropy", "probmax"]
-        ) and args.entropy_model_checkpoint_dir is None
 
         self.apply_transformer = args.use_local_encoder_transformer
         self.downsampling_by_pooling = args.downsampling_by_pooling