deepgeek/MiniCPM-o
2
0
Fork 0
mirror of https://github.com/OpenBMB/MiniCPM-V.git synced 2026-02-04 17:59:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update inference_on_multiple_gpus.md

Browse Source 
typos
This commit is contained in:
ByeongkiJeong
2024-06-04 09:55:52 +09:00
committed by GitHub
parent e95b488cfe
commit 131ae3d294
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/inference_on_multiple_gpus.md
View File

@@ -43,7 +43,7 @@ gpu_device_ids = [0, 1] # Define which gpu to use (now we have two GPUs, each ha
no_split_module_classes = ["LlamaDecoderLayer"]
max_memory = {
device_id: memory for device_id in gpu_device_ids
device_id: max_memory_each_gpu for device_id in gpu_device_ids
}
config = AutoConfig.from_pretrained(