diff --git a/QEfficient/transformers/models/qwen2_5_vl/modeling_qwen2_5_vl.py b/QEfficient/transformers/models/qwen2_5_vl/modeling_qwen2_5_vl.py index 7944dff653..45c6616018 100644 --- a/QEfficient/transformers/models/qwen2_5_vl/modeling_qwen2_5_vl.py +++ b/QEfficient/transformers/models/qwen2_5_vl/modeling_qwen2_5_vl.py @@ -1091,7 +1091,7 @@ def get_specializations( "resolution." ) else: - assert vision_size * f < user_vision_size, ( + assert vision_size * f <= user_vision_size, ( f"Computed vision_size of {vision_size * f} tokens " f"(vision_size={vision_size}, num_frames={f}) for image resolution " f"(width={w}, height={h}) cannot exceed the provided " diff --git a/QEfficient/transformers/models/qwen3_vl/modeling_qwen3_vl.py b/QEfficient/transformers/models/qwen3_vl/modeling_qwen3_vl.py index c914d48c3e..2d834423f6 100644 --- a/QEfficient/transformers/models/qwen3_vl/modeling_qwen3_vl.py +++ b/QEfficient/transformers/models/qwen3_vl/modeling_qwen3_vl.py @@ -972,7 +972,7 @@ def get_specializations( "resolution." ) else: - assert vision_size * f < user_vision_size, ( + assert vision_size * f <= user_vision_size, ( f"Computed vision_size of {vision_size * f} tokens " f"(vision_size={vision_size}, num_frames={f}) for image resolution " f"(width={w}, height={h}) cannot exceed the provided " diff --git a/QEfficient/transformers/models/qwen3_vl_moe/modeling_qwen3_vl_moe.py b/QEfficient/transformers/models/qwen3_vl_moe/modeling_qwen3_vl_moe.py index 83cf3d40e8..078cb4afb7 100644 --- a/QEfficient/transformers/models/qwen3_vl_moe/modeling_qwen3_vl_moe.py +++ b/QEfficient/transformers/models/qwen3_vl_moe/modeling_qwen3_vl_moe.py @@ -1001,7 +1001,7 @@ def get_specializations( "resolution." ) else: - assert vision_size * f < user_vision_size, ( + assert vision_size * f <= user_vision_size, ( f"Computed vision_size of {vision_size * f} tokens " f"(vision_size={vision_size}, num_frames={f}) for image resolution " f"(width={w}, height={h}) cannot exceed the provided "