feat(M2.1): 设备端部署脚本 + whisper LD_LIBRARY_PATH + dog/cat 角色人设配置
This commit is contained in:
@@ -47,10 +47,15 @@ impl AiPlugin {
|
||||
pub fn new_default(
|
||||
model_store: PathBuf,
|
||||
tools_dir: PathBuf,
|
||||
whisper_lib_dir: Option<PathBuf>,
|
||||
tmp_dir: PathBuf,
|
||||
context_window: usize,
|
||||
) -> Self {
|
||||
let backend = Box::new(backend::LocalCliBackend::new(tools_dir.clone()));
|
||||
let mut backend_builder = backend::LocalCliBackend::new(tools_dir.clone());
|
||||
if let Some(lib_dir) = whisper_lib_dir {
|
||||
backend_builder = backend_builder.with_whisper_lib_dir(lib_dir);
|
||||
}
|
||||
let backend = Box::new(backend_builder);
|
||||
let models = model_manager::ModelManager::new(model_store);
|
||||
let pipeline = ChatPipeline::new(backend, models, tmp_dir, context_window);
|
||||
Self {
|
||||
|
||||
Reference in New Issue
Block a user