remove unused argument

This commit is contained in:
Luciferian Ink 2025-01-17 18:04:25 -06:00
parent 420326184a
commit 6129756e10

View file

@ -199,9 +199,6 @@ class LocalModelBase(nn.Module):
class LocalEncoder(LocalModelBase): class LocalEncoder(LocalModelBase):
def __init__(self, args: LocalModelArgs): def __init__(self, args: LocalModelArgs):
super().__init__(args) 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.apply_transformer = args.use_local_encoder_transformer
self.downsampling_by_pooling = args.downsampling_by_pooling self.downsampling_by_pooling = args.downsampling_by_pooling