Skip to content

Commit 8cd9043

Browse files
committed
Revert "Guard Qwen3VL imports with try/except in transformers_models.py"
This reverts commit 10038f0.
1 parent 10038f0 commit 8cd9043

1 file changed

Lines changed: 2 additions & 9 deletions

File tree

tests/_test_utils/torch/transformers_models.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -32,16 +32,11 @@
3232
PreTrainedModel,
3333
Qwen3Config,
3434
Qwen3MoeConfig,
35+
Qwen3VLConfig,
3536
T5Config,
3637
T5ForConditionalGeneration,
3738
)
38-
39-
try:
40-
from transformers import Qwen3VLConfig
41-
from transformers.models.qwen3_vl.modeling_qwen3_vl import Qwen3VLForConditionalGeneration
42-
except ImportError:
43-
Qwen3VLConfig = None # type: ignore[assignment,misc]
44-
Qwen3VLForConditionalGeneration = None # type: ignore[assignment,misc]
39+
from transformers.models.qwen3_vl.modeling_qwen3_vl import Qwen3VLForConditionalGeneration
4540

4641
import modelopt.torch.opt as mto
4742

@@ -130,8 +125,6 @@ def create_tiny_qwen3_moe_dir(
130125

131126
##### Qwen3-VL #####
132127
def get_tiny_qwen3vl(**config_kwargs) -> PreTrainedModel:
133-
if Qwen3VLConfig is None:
134-
pytest.skip("transformers does not have Qwen3VL support")
135128
set_seed(SEED)
136129

137130
# Defaults: hidden_size=num_attention_heads*head_dim (e.g. 4*8=32).

0 commit comments

Comments
 (0)