Skip to content

Commit 80bee7b

Browse files
authored
Avoid many test failures for LlavaNextVideoForConditionalGeneration (#34070)
* skip * [run-slow] llava_next_video * skip * [run-slow] video_llava, llava_next_video * skip * [run-slow] llava_next_video --------- Co-authored-by: ydshieh <[email protected]>
1 parent 37ac078 commit 80bee7b

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/generation/test_utils.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1230,6 +1230,9 @@ def test_dola_decoding_sample(self):
12301230
if any(model_name in model_class.__name__.lower() for model_name in ["marian", "mbart", "pegasus"]):
12311231
self.skipTest("DoLa is not supported for models that don't return layerwise hidden states")
12321232

1233+
if any(model_name == model_class.__name__ for model_name in ["LlavaNextVideoForConditionalGeneration"]):
1234+
self.skipTest(f"DoLa is failing for {model_class.__name__}")
1235+
12331236
# enable cache if the model is not openai-gpt, xlnet, cpm, or xlm
12341237
config, inputs_dict = self.prepare_config_and_inputs_for_generate()
12351238
main_input = inputs_dict[model_class.main_input_name]

0 commit comments

Comments
 (0)