From 28a7df85196cbcae26e44b3f59ceaa25b52bb32a Mon Sep 17 00:00:00 2001 From: haixuanTao Date: Wed, 11 Sep 2024 12:28:38 +0200 Subject: [PATCH] Fix pylint --- node-hub/dora-qwenvl/dora_qwenvl/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/node-hub/dora-qwenvl/dora_qwenvl/main.py b/node-hub/dora-qwenvl/dora_qwenvl/main.py index d659b4df2..92294b486 100644 --- a/node-hub/dora-qwenvl/dora_qwenvl/main.py +++ b/node-hub/dora-qwenvl/dora_qwenvl/main.py @@ -15,7 +15,7 @@ # Check if flash_attn is installed try: - import flash_attn + import flash_attn as _ model = Qwen2VLForConditionalGeneration.from_pretrained( CUSTOM_MODEL_PATH,