File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 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
4641import modelopt .torch .opt as mto
4742
@@ -130,8 +125,6 @@ def create_tiny_qwen3_moe_dir(
130125
131126##### Qwen3-VL #####
132127def 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).
You can’t perform that action at this time.
0 commit comments