Update DeepSeek-V3-Chat-multi-gpu-marlin.yaml

This commit is contained in:
Atream 2025-02-26 21:53:50 +08:00 committed by GitHub
parent 9660b2cc1e
commit 90eb87b3fc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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"