Browse Source

Fix linting

tags/v0.3.7rc0
haixuanTao 1 year ago
parent
commit
33307b4e8a
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

@@ -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",


Loading…
Cancel
Save