Apply suggestions from code review

huggingface · Feb 15, 2024 · 5f54d84 · 5f54d84
1 parent c9f3c82
commit 5f54d84
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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