Fix flaky tests (#34069)

* fix mllama only * allow image token index
2024-10-11 15:41:46 +02:00
parent 3a24ba82ad
commit fd70464fa7
2 changed files with 2 additions and 1 deletions
--- a/utils/check_config_attributes.py
+++ b/utils/check_config_attributes.py
@@ -243,6 +243,7 @@ def check_attribute_being_used(config_class, attributes, default_value, source_s
        "pad_index",
        "unk_index",
        "mask_index",
+        "image_token_index",  # for VLMs
        "image_size",
        "use_cache",
        "out_features",