diff --git a/tests/structs/test_model_parallizer.py b/tests/structs/test_model_parallizer.py index a0840608..2966712e 100644 --- a/tests/structs/test_model_parallizer.py +++ b/tests/structs/test_model_parallizer.py @@ -15,7 +15,7 @@ custom_config = { "verbose": True, } huggingface_llm = HuggingfaceLLM( - model_id="NousResearch/Nous-Hermes-2-Vision-Alpha", + model_id="liuhaotian/llava-v1.6-mistral-7b", **custom_config, ) mixtral = Mixtral(load_in_4bit=True, use_flash_attention_2=True)