Browse Source

Fix pylint

tags/v0.3.7rc0
haixuanTao 1 year ago
parent
commit
28a7df8519
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      node-hub/dora-qwenvl/dora_qwenvl/main.py

+ 1
- 1
node-hub/dora-qwenvl/dora_qwenvl/main.py View File

@@ -15,7 +15,7 @@ DEFAULT_QUESTION = os.getenv(

# Check if flash_attn is installed
try:
import flash_attn
import flash_attn as _

model = Qwen2VLForConditionalGeneration.from_pretrained(
CUSTOM_MODEL_PATH,


Loading…
Cancel
Save