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,