vlm: optimize tensor transport (#6003)

Co-authored-by: Xinyuan Tong <115166877+JustinTong0323@users.noreply.github.com>
This commit is contained in:
Mick
2025-07-26 17:41:01 +08:00
committed by GitHub
parent 534756749a
commit 3212c2ad3f
23 changed files with 221 additions and 60 deletions

View File

@@ -24,7 +24,7 @@ class VLMInputTestBase:
model_path = None
chat_template = None
processor = None
visual = None # Should be a callable for precomputed features
visual = None # Should be a callable for precomputed embeddings
@classmethod
def setUpClass(cls):
@@ -41,7 +41,7 @@ class VLMInputTestBase:
@classmethod
def _init_visual(cls):
"""Override in subclass to set up cls.visual as a callable for precomputed features."""
"""Override in subclass to set up cls.visual as a callable for precomputed embeddings."""
raise NotImplementedError
def setUp(self):