You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
# int32_t n_gpu_layers; // number of layers to store in VRAM
684
+
# int32_t n_gpu_layers; // number of layers to store in VRAM, a negative value means all layers
685
685
# enum llama_split_mode split_mode; // how to split the model across multiple GPUs
686
686
687
687
# // main_gpu interpretation depends on split_mode:
@@ -720,7 +720,7 @@ class llama_model_params(ctypes.Structure):
720
720
Attributes:
721
721
devices (ctypes.Array[ggml_backend_dev_t]): NULL-terminated list of devices to use for offloading (if NULL, all available devices are used)
722
722
tensor_buft_overrides(llama_model_tensor_buft_override): NULL-terminated list of buffer types to use for tensors that match a pattern
723
-
n_gpu_layers (int): number of layers to store in VRAM
723
+
n_gpu_layers (int): number of layers to store in VRAM, a negative value means all layers
724
724
split_mode (int): how to split the model across multiple GPUs
725
725
main_gpu (int): the GPU that is used for the entire model. main_gpu interpretation depends on split_mode: LLAMA_SPLIT_NONE: the GPU that is used for the entire model LLAMA_SPLIT_ROW: the GPU that is used for small tensors and intermediate results LLAMA_SPLIT_LAYER: ignored
726
726
tensor_split (ctypes.Array[ctypes.ctypes.c_float]): proportion of the model (layers or rows) to offload to each GPU, size: llama_max_devices()
0 commit comments