From 64d9b62e26259b5873a3bc55b1a4c5ce08841c37 Mon Sep 17 00:00:00 2001 From: Yu xing Date: Wed, 21 Aug 2024 17:16:48 -0700 Subject: [PATCH] remove vlm test --- tests/vlm_test.py | 25 ------------------------- 1 file changed, 25 deletions(-) delete mode 100644 tests/vlm_test.py diff --git a/tests/vlm_test.py b/tests/vlm_test.py deleted file mode 100644 index b7a450f2..00000000 --- a/tests/vlm_test.py +++ /dev/null @@ -1,25 +0,0 @@ -from nexa.gguf import NexaVLMInference -import tempfile -from .utils import download_model - -def test_image_generation(): - with tempfile.TemporaryDirectory() as temp_dir: - model = NexaVLMInference( - model_path="nanollava", - ) - image_path = download_model( - "https://www.wikipedia.org/portal/wikipedia.org/assets/img/Wikipedia-logo-v2.png", - temp_dir, - ) - output = model._chat("what's in this image?", image_path) - for chunk in output: - delta = chunk["choices"][0]["delta"] - if "role" in delta: - print(delta["role"], end=": ") - elif "content" in delta: - print(delta["content"], end="") - - -# if __name__ == "__main__": -# print("=== Testing 1 ===") -# test1()