Fix pylint

dora-rs · Sep 11, 2024 · 28a7df8 · 28a7df8
1 parent 39f3fd4
commit 28a7df8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,