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,