|
1 | 1 | //! A safe wrapper around `llama_model_params`. |
2 | 2 |
|
3 | 3 | use crate::model::params::kv_overrides::KvOverrides; |
4 | | -use crate::LLamaCppError; |
| 4 | +use crate::LlamaCppError; |
5 | 5 | use std::ffi::{c_char, CStr}; |
6 | 6 | use std::fmt::{Debug, Formatter}; |
7 | 7 | use std::pin::Pin; |
@@ -375,19 +375,19 @@ impl LlamaModelParams { |
375 | 375 | /// You don't need to specify CPU or ACCEL devices. |
376 | 376 | /// |
377 | 377 | /// # Errors |
378 | | - /// Returns `LLamaCppError::BackendDeviceNotFound` if any device index is invalid. |
379 | | - pub fn with_devices(mut self, devices: &[usize]) -> Result<Self, LLamaCppError> { |
| 378 | + /// Returns `LlamaCppError::BackendDeviceNotFound` if any device index is invalid. |
| 379 | + pub fn with_devices(mut self, devices: &[usize]) -> Result<Self, LlamaCppError> { |
380 | 380 | for dev in self.devices.iter_mut() { |
381 | 381 | *dev = std::ptr::null_mut(); |
382 | 382 | } |
383 | 383 | // Check device count |
384 | 384 | let max_devices = crate::max_devices().min(LLAMA_CPP_MAX_DEVICES); |
385 | 385 | if devices.len() > max_devices { |
386 | | - return Err(LLamaCppError::MaxDevicesExceeded(max_devices)); |
| 386 | + return Err(LlamaCppError::MaxDevicesExceeded(max_devices)); |
387 | 387 | } |
388 | 388 | for (i, &dev) in devices.iter().enumerate() { |
389 | 389 | if dev >= unsafe { llama_cpp_sys_2::ggml_backend_dev_count() } { |
390 | | - return Err(LLamaCppError::BackendDeviceNotFound(dev)); |
| 390 | + return Err(LlamaCppError::BackendDeviceNotFound(dev)); |
391 | 391 | } |
392 | 392 | let backend_dev = unsafe { llama_cpp_sys_2::ggml_backend_dev_get(dev) }; |
393 | 393 | self.devices[i] = backend_dev; |
|
0 commit comments