From 33307b4e8a5460d038c9cca6ebfaed76b4a04b37 Mon Sep 17 00:00:00 2001 From: haixuanTao Date: Mon, 7 Oct 2024 11:18:06 +0200 Subject: [PATCH] Fix linting --- 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 1075a177..e03b31c1 100644 --- a/node-hub/dora-qwenvl/dora_qwenvl/main.py +++ b/node-hub/dora-qwenvl/dora_qwenvl/main.py @@ -24,7 +24,7 @@ try: device_map="auto", attn_implementation="flash_attention_2", ) -except ImportError or ModuleNotFoundError: +except (ImportError, ModuleNotFoundError): model = Qwen2VLForConditionalGeneration.from_pretrained( CUSTOM_MODEL_PATH, torch_dtype="auto",