diff --git a/torchmultimodal/modules/encoders/mil_encoder.py b/torchmultimodal/modules/encoders/mil_encoder.py
index 67aa4a42..c86962f1 100644
--- a/torchmultimodal/modules/encoders/mil_encoder.py
+++ b/torchmultimodal/modules/encoders/mil_encoder.py
@@ -83,7 +83,6 @@ def get_channel_name(self, id: int) -> str:
         return f"mil_{id}"
 
     def forward(self, x: Tensor) -> Tensor:
-        partitioned_input = {}
         idx = 0
         input_size = x.size(dim=1)
         if input_size != sum(self.partition_sizes):