From 5f54d84e3f291ea636ecc9d71061b901063df958 Mon Sep 17 00:00:00 2001 From: Arthur <48595927+ArthurZucker@users.noreply.github.com> Date: Thu, 15 Feb 2024 01:21:14 +0100 Subject: [PATCH] Apply suggestions from code review --- src/transformers/models/llama/modeling_llama.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/transformers/models/llama/modeling_llama.py b/src/transformers/models/llama/modeling_llama.py index ddbaffd1477d0e..8cc81e6ad7d54e 100644 --- a/src/transformers/models/llama/modeling_llama.py +++ b/src/transformers/models/llama/modeling_llama.py @@ -1013,7 +1013,6 @@ def forward( all_hidden_states += (hidden_states,) next_cache = None - # if use_cache and isinstance(next_decoder_cache, (DynamicCache, SinkCache)): if use_cache: next_cache = ( next_decoder_cache.to_legacy_cache() if isinstance(next_decoder_cache, Cache) else next_decoder_cache