RYDE-WORK/ktransformers
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/ktransformers.git synced 2026-01-19 12:43:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #697 from kvcache-ai/fix-yaml

Browse Source 
Update DeepSeek-V3-Chat-multi-gpu-marlin.yaml
This commit is contained in:
Atream 2025-02-26 21:54:01 +08:00 committed by GitHub
commit 9650893adc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ktransformers/optimize/optimize_rules/DeepSeek-V3-Chat-multi-gpu-marlin.yaml
View File

@ -168,5 +168,5 @@
replace:
class: "default"
kwargs:
generate_device: "cuda:0"
prefill_device: "cuda:0"
generate_device: "cuda:1"
prefill_device: "cuda:1"