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
17b17c06
Unverified
Commit
17b17c06
authored
Jan 29, 2026
by
sthWrong
Committed by
GitHub
Jan 29, 2026
Browse files
[Backport] [Kimi-K2.5] Replace torch.cuda with current_platform for d… (#33320)
parent
8bb6271c
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
1 deletion
+2
-1
vllm/model_executor/models/kimi_k25.py
vllm/model_executor/models/kimi_k25.py
+2
-1
No files found.
vllm/model_executor/models/kimi_k25.py
View file @
17b17c06
...
...
@@ -58,6 +58,7 @@ from vllm.multimodal.processing import (
PromptReplacement
,
PromptUpdate
,
)
from
vllm.platforms
import
current_platform
from
vllm.sequence
import
IntermediateTensors
from
vllm.transformers_utils.configs
import
KimiK25Config
from
vllm.transformers_utils.processor
import
cached_get_image_processor
...
...
@@ -320,7 +321,7 @@ class KimiK25ForConditionalGeneration(nn.Module, SupportsMultiModal, SupportsPP)
model_config
.
multimodal_config
.
mm_encoder_tp_mode
==
"data"
)
self
.
hidden_size
=
config
.
text_config
.
hidden_size
self
.
device
=
torch
.
cuda
.
current_device
()
self
.
device
=
current_platform
.
current_device
()
# Build vision tower directly with KimiK25VisionConfig
self
.
vision_tower
=
MoonViT3dPretrainedModel
(
config
.
vision_config
,
...
...
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