Remove internvl

Author: tomasruizt

llmlib/llmlib/internvl.py

@@ -1,5 +1,4 @@
 from .replicate_api import Apollo7B
-from .internvl import InternVL
 from .gemini.gemini_code import GeminiAPI
 from .gemma import PaliGemma2
 from .minicpm import MiniCPM
@@ -13,7 +12,6 @@ def filled_model_registry() -> ModelRegistry:
     return ModelRegistry(
         models=[
             *model_entries_from_mult_ids(MiniCPM),
-            ModelEntry.from_cls_with_id(InternVL),
             ModelEntry.from_cls_with_id(Apollo7B),
             ModelEntry.from_cls_with_id(Phi3Vision),
             ModelEntry.from_cls_with_id(LLama3Vision8B),

llmlib/llmlib/runtime.py

@@ -11,7 +11,6 @@ dependencies = [
     "bugsnag>=4.7.1",
     "google-cloud-aiplatform>=1.64",
     "decorator>=5.1.1",
-    "timm>=1.0.12",  # for internvl2.5
     "replicate>=1.0.4",
     "transformers>=4.47.1",
 ]

llmlib/pyproject.toml

@@ -1,13 +1,11 @@
 from io import BytesIO
-from llmlib.internvl import InternVL
 from llmlib.minicpm import MiniCPM, to_listof_imgs
 from llmlib.replicate_api import Apollo7B
 import pytest
 from .helpers import (
     assert_model_knows_capital_of_france,
     assert_model_recognizes_afd_in_video,
     assert_model_recognizes_pyramid_in_image,
-    assert_model_supports_multiturn,
     file_for_test,
     is_ci,
 )
@@ -21,14 +19,6 @@ def test_minicpm_vision():
     assert_model_recognizes_afd_in_video(model)
 
 
-@pytest.mark.skipif(condition=is_ci(), reason="No GPU in CI")
-def test_internvl():
-    model = InternVL()
-    assert_model_knows_capital_of_france(model)
-    assert_model_recognizes_pyramid_in_image(model)
-    assert_model_supports_multiturn(model)
-
-
 @pytest.mark.skipif(condition=is_ci(), reason="No GPU in CI")
 def test_apollo_7b():
     model = Apollo7B()