From 103c6616c4c5e58061864af52b8a1bb988a73be2 Mon Sep 17 00:00:00 2001 From: Shunta Saito Date: Fri, 14 Feb 2025 23:06:42 +0900 Subject: [PATCH] Remove unused code part --- llms/mlx_lm/models/plamo2.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/llms/mlx_lm/models/plamo2.py b/llms/mlx_lm/models/plamo2.py index 7edfef07..f0f5d25b 100644 --- a/llms/mlx_lm/models/plamo2.py +++ b/llms/mlx_lm/models/plamo2.py @@ -1648,9 +1648,6 @@ class Model(PlamoPreTrainedModel): raise ValueError( f"Unexpected output type for causal language model: {type(output)} != CausalLMOutputWithPast" ) - if output.past_key_values is not None: - # output.past_key_values is actually a PlamoCache object - self._cache = output.past_key_values # type: ignore if output.logits is not None: return output.logits else: