tadkt
/

GOT_Vietnamese

Image-Text-to-Text

feature-extraction

vision-language

Model card Files Files and versions

Update modeling_GOT.py

#1

by htrnguyen - opened May 17, 2025

base: refs/heads/main

←

from: refs/pr/1

Discussion Files changed

Files changed (1) hide show

modeling_GOT.py +1 -1

modeling_GOT.py CHANGED Viewed

@@ -398,7 +398,7 @@ class GOTQwenForCausalLM(Qwen2ForCausalLM):
             if isinstance(past_key_values, Cache):
                 cache_length = past_key_values.get_seq_length()
                 past_length = past_key_values.seen_tokens
-                max_cache_length = past_key_values.get_max_length()
             else:
                 cache_length = past_length = past_key_values[0][0].shape[2]
                 max_cache_length = None

             if isinstance(past_key_values, Cache):
                 cache_length = past_key_values.get_seq_length()
                 past_length = past_key_values.seen_tokens
+                max_cache_length = None
             else:
                 cache_length = past_length = past_key_values[0][0].shape[2]
                 max_cache_length = None