Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
vllm_cscc
Commits
a85724bd
Unverified
Commit
a85724bd
authored
Dec 18, 2025
by
wangxiyuan
Committed by
GitHub
Dec 18, 2025
Browse files
[Platform] Let EPD work with non-cuda platform (#30225)
Signed-off-by:
wangxiyuan
<
wangxiyuan1007@gmail.com
>
parent
11a89cf9
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
1 deletion
+4
-1
vllm/distributed/ec_transfer/ec_connector/example_connector.py
...distributed/ec_transfer/ec_connector/example_connector.py
+4
-1
No files found.
vllm/distributed/ec_transfer/ec_connector/example_connector.py
View file @
a85724bd
...
...
@@ -73,6 +73,7 @@ class ECExampleConnector(ECConnectorBase):
data hashes (`mm_hash`) to encoder cache tensors.
kwargs (dict): Additional keyword arguments for the connector.
"""
from
vllm.platforms
import
current_platform
# Get the metadata
metadata
:
ECConnectorMetadata
=
self
.
_get_connector_metadata
()
...
...
@@ -91,7 +92,9 @@ class ECExampleConnector(ECConnectorBase):
if
mm_data
.
mm_hash
in
encoder_cache
:
continue
filename
=
self
.
_generate_filename_debug
(
mm_data
.
mm_hash
)
ec_cache
=
safetensors
.
torch
.
load_file
(
filename
)[
"ec_cache"
].
cuda
()
ec_cache
=
safetensors
.
torch
.
load_file
(
filename
,
device
=
current_platform
.
device_type
)[
"ec_cache"
]
encoder_cache
[
mm_data
.
mm_hash
]
=
ec_cache
logger
.
debug
(
"Success load encoder cache for hash %s"
,
mm_data
.
mm_hash
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment