refactor: 优化 ModelLoader 结构与维度解析逻辑
- 提取 `resolve_tensors` 公共函数,消除输入输出流的重复代码。 - 简化 `resolve_shape`,使其回归纯粹的维度翻译职责,移除过早的错误校验。 - 统一错误处理,将底层解析异常清晰映射至 `DdddError::InternalError`。
This commit is contained in:
@@ -17,7 +17,6 @@ pub(crate) const MODEL_DOWNLOAD_HELP: &str = "\
|
||||
B. 或者直接将模型文件重命名并放置在您运行程序的“当前工作目录”或“可执行文件同级目录”下。
|
||||
================================================================================";
|
||||
|
||||
|
||||
use thiserror::Error;
|
||||
|
||||
#[derive(Error, Debug)]
|
||||
@@ -42,7 +41,10 @@ pub enum DdddError {
|
||||
|
||||
#[error("未知的模型输出格式")]
|
||||
UnknownOutputFormat,
|
||||
|
||||
#[error("解析节点 Fact 失败")]
|
||||
InternalError(String),
|
||||
}
|
||||
|
||||
/// 统一用我们自己的 DdddError 包装 Result
|
||||
pub type Result<T> = std::result::Result<T, DdddError>;
|
||||
pub type Result<T> = std::result::Result<T, DdddError>;
|
||||
|
||||
Reference in New Issue
Block a user