[CI/Build] Fix model nightly tests

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>

Author: DarkLight1337

tests/models/language/generation/test_common.py

@@ -100,7 +100,7 @@
             "allenai/OLMoE-1B-7B-0924-Instruct",
             marks=[pytest.mark.cpu_model],
         ),
-        pytest.param("swiss-ai/Apertus-8B-2509"),  # apertus
+        pytest.param("swiss-ai/Apertus-8B-Instruct-2509"),  # apertus
     ],
 )
 @pytest.mark.parametrize("max_tokens", [32])

tests/models/multimodal/generation/test_common.py

@@ -749,6 +749,7 @@
         max_num_seqs=2,
         auto_cls=AutoModelForImageTextToText,
         hf_output_post_proc=model_utils.smolvlm_trunc_hf_output,
+        num_logprobs=10,
     ),
     "tarsier": VLMTestInfo(
         models=["omni-research/Tarsier-7b"],

tests/models/multimodal/pooling/test_clip.py

@@ -45,14 +45,16 @@ def _run_test(
 
         all_outputs = []
         for inputs in all_inputs:
+            inputs = hf_model.wrap_device(inputs)
+
             if "pixel_values" in inputs:
-                inputs.pop("input_ids")
                 pooled_output = hf_model.model.get_image_features(
-                    **hf_model.wrap_device(inputs)
+                    pixel_values=inputs.pixel_values,
                 ).squeeze(0)
             else:
                 pooled_output = hf_model.model.get_text_features(
-                    **hf_model.wrap_device(inputs)
+                    input_ids=inputs.input_ids,
+                    attention_mask=inputs.attention_mask,
                 ).squeeze(0)
 
             all_outputs.append(pooled_output.tolist())

tests/models/registry.py

@@ -172,9 +172,8 @@ def check_available_online(
 _TEXT_GENERATION_EXAMPLE_MODELS = {
     # [Decoder-only]
     "ApertusForCausalLM": _HfExamplesInfo(
-        "swiss-ai/Apertus-8B-2509",
+        "swiss-ai/Apertus-8B-Instruct-2509",
         min_transformers_version="4.56.0",
-        trust_remote_code=True,
     ),
     "AquilaModel": _HfExamplesInfo("BAAI/AquilaChat-7B", trust_remote_code=True),
     "AquilaForCausalLM": _HfExamplesInfo("BAAI/AquilaChat2-7B", trust_remote_code=True),