From eabcb2967a21392a7ced328b7f6d24e88efc1843 Mon Sep 17 00:00:00 2001 From: matvey-kolbasov-hs Date: Wed, 24 Jul 2024 15:37:16 +0300 Subject: [PATCH] logging --- server/text_generation_server/models/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/server/text_generation_server/models/__init__.py b/server/text_generation_server/models/__init__.py index a1ccdb12..ab6af350 100644 --- a/server/text_generation_server/models/__init__.py +++ b/server/text_generation_server/models/__init__.py @@ -950,6 +950,7 @@ def get_model( if model_type == QWEN2: if FLASH_ATTENTION: + print('!!! aliases !!!') return FlashCausalLM( model_id=model_id, model_class=Qwen2ForCausalLM,