From 6bfddee5a3ac37d98631cb91fe2fd28c91ec395f Mon Sep 17 00:00:00 2001 From: vyomakesh09 Date: Fri, 29 Dec 2023 00:39:35 +0000 Subject: [PATCH] modified: tests/swarms/test_model_parallizer.py --- tests/swarms/test_model_parallizer.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/tests/swarms/test_model_parallizer.py b/tests/swarms/test_model_parallizer.py index e9a85c9f..fb3dc2b2 100644 --- a/tests/swarms/test_model_parallizer.py +++ b/tests/swarms/test_model_parallizer.py @@ -3,7 +3,6 @@ from swarms.swarms.model_parallizer import ModelParallelizer from swarms.models import ( HuggingfaceLLM, Mixtral, - Dalle3, GPT4VisionAPI, ZeroscopeTTV, ) @@ -19,7 +18,6 @@ huggingface_llm = HuggingfaceLLM( **custom_config, ) mixtral = Mixtral(load_in_4bit=True, use_flash_attention_2=True) -dalle3 = Dalle3(quality="high") gpt4_vision_api = GPT4VisionAPI(max_tokens=1000) zeroscope_ttv = ZeroscopeTTV() @@ -29,7 +27,6 @@ def test_init(): [ huggingface_llm, mixtral, - dalle3, gpt4_vision_api, zeroscope_ttv, ] @@ -51,7 +48,6 @@ def test_run_all(): [ huggingface_llm, mixtral, - dalle3, gpt4_vision_api, zeroscope_ttv, ]