Skip to content

Commit 3468ae7

Browse files
committed
Implement Default for LlamaSplitMode
1 parent d364092 commit 3468ae7

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

llama-cpp-2/src/model/params.rs

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,13 @@ impl From<LlamaSplitMode> for i32 {
5454
}
5555
}
5656

57+
/// The default split mode is `Layer` in llama.cpp.
58+
impl Default for LlamaSplitMode {
59+
fn default() -> Self {
60+
LlamaSplitMode::Layer
61+
}
62+
}
63+
5764
/// The maximum number of devices supported.
5865
///
5966
/// The real maximum number of devices is the lesser one of this value and the value returned by

0 commit comments

Comments
 (0)