fix

Signed-off-by: Pamela <[email protected]>

Author: pamelap-nvidia

tensorrt_llm/_torch/models/modeling_qwen2vl.py

@@ -4,8 +4,9 @@
 
 import torch
 import torch.nn as nn
-from transformers import (AutoProcessor, AutoTokenizer, PretrainedConfig,
-                          PreTrainedModel, Qwen2_5_VLForConditionalGeneration,
+from transformers import (AutoConfig, AutoProcessor, AutoTokenizer,
+                          PretrainedConfig, PreTrainedModel,
+                          Qwen2_5_VLForConditionalGeneration,
                           Qwen2VLForConditionalGeneration)
 from transformers.models.qwen2_5_vl.modeling_qwen2_5_vl import \
     Qwen2_5_VisionTransformerPretrainedModel