From db0b6567e1d2121e0f19f566c22ecf7c6ce4dccd Mon Sep 17 00:00:00 2001 From: yuanwu Date: Mon, 29 Jul 2024 22:02:42 +0000 Subject: [PATCH] Remove log Signed-off-by: yuanwu --- server/text_generation_server/models/__init__.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/server/text_generation_server/models/__init__.py b/server/text_generation_server/models/__init__.py index 569b204f..30bd8f6c 100644 --- a/server/text_generation_server/models/__init__.py +++ b/server/text_generation_server/models/__init__.py @@ -165,9 +165,8 @@ def get_model( dtype=dtype, trust_remote_code=trust_remote_code, ) - logger.info(f"model_type = {model_type}") + if model_type == "llava_next": - logger.info(f"################model_type = {model_type}") return VlmCausalLM( model_class=LlavaNextForConditionalGeneration, model_id=model_id,