Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 9 additions & 0 deletions tests/models/multimodal/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,18 @@
import warnings

import torch
import torch.nn.functional as F

from vllm.platforms import current_platform

# openbmb/MiniCPM-Llama3-V-2_5's remote ``resampler.py`` annotates with ``List``
# without importing it, relying on the alias leaking from
# ``from torch.nn.functional import *``. Recent torch uses builtin generics and no
# longer re-exports the typing aliases, so executing that module raises NameError.
# TODO: Remove once the checkpoint fixes its imports.
if not hasattr(F, "List"):
F.List = list


def pytest_configure(config):
"""Early ROCm configuration that must happen before test collection."""
Expand Down
Loading