mirror of
https://github.com/neodyland/sbv2-api.git
synced 2026-01-05 05:52:56 +00:00
fix: agpl issue
This commit is contained in:
12
.github/workflows/CI.yml
vendored
12
.github/workflows/CI.yml
vendored
@@ -201,9 +201,9 @@ jobs:
|
||||
|
||||
- name: Merge
|
||||
run: |
|
||||
docker buildx imagetools create -t ${{ env.GHCR_REPO }}:cuda \
|
||||
${{ env.GHCR_REPO }}:latest-cuda-amd64 \
|
||||
${{ env.GHCR_REPO }}:latest-cuda-arm64
|
||||
docker buildx imagetools create -t ${{ env.GHCR_REPO }}:cpu \
|
||||
${{ env.GHCR_REPO }}:latest-cpu-amd64 \
|
||||
${{ env.GHCR_REPO }}:latest-cpu-arm64
|
||||
docker buildx imagetools create -t ghcr.io/${{ github.repository }}:cuda \
|
||||
ghcr.io/${{ github.repository }}:latest-cuda-amd64 \
|
||||
ghcr.io/${{ github.repository }}:latest-cuda-arm64
|
||||
docker buildx imagetools create -t ghcr.io/${{ github.repository }}:cpu \
|
||||
ghcr.io/${{ github.repository }}:latest-cpu-amd64 \
|
||||
ghcr.io/${{ github.repository }}:latest-cpu-arm64
|
||||
|
||||
@@ -14,7 +14,7 @@ axum = "0.8.0"
|
||||
dotenvy.workspace = true
|
||||
env_logger.workspace = true
|
||||
log = "0.4.22"
|
||||
sbv2_core = { version = "0.2.0-alpha2", path = "../sbv2_core", features = ["aivmx"] }
|
||||
sbv2_core = { version = "0.2.0-alpha6", path = "../sbv2_core", features = ["aivmx"] }
|
||||
serde = { version = "1.0.210", features = ["derive"] }
|
||||
tokio = { version = "1.40.0", features = ["full"] }
|
||||
utoipa = { version = "5.0.0", features = ["axum_extras"] }
|
||||
|
||||
@@ -17,4 +17,8 @@ crate-type = ["cdylib"]
|
||||
anyhow.workspace = true
|
||||
ndarray.workspace = true
|
||||
pyo3 = { version = "0.23.0", features = ["anyhow"] }
|
||||
sbv2_core = { version = "0.2.0-alpha2", path = "../sbv2_core" }
|
||||
sbv2_core = { path = "../sbv2_core", features = ["std"], default-features = false }
|
||||
|
||||
[features]
|
||||
agpl_dict = ["sbv2_core/agpl_dict"]
|
||||
default = ["agpl_dict"]
|
||||
@@ -31,12 +31,13 @@ zstd = "0.13.2"
|
||||
[features]
|
||||
cuda = ["ort/cuda", "std"]
|
||||
cuda_tf32 = ["std", "cuda"]
|
||||
agpl_dict = []
|
||||
std = ["dep:ort", "tokenizers/progressbar", "tokenizers/onig", "tokenizers/esaxx_fast"]
|
||||
dynamic = ["ort/load-dynamic", "std"]
|
||||
directml = ["ort/directml", "std"]
|
||||
tensorrt = ["ort/tensorrt", "std"]
|
||||
coreml = ["ort/coreml", "std"]
|
||||
default = ["std"]
|
||||
default = ["std", "agpl_dict"]
|
||||
no_std = ["tokenizers/unstable_wasm"]
|
||||
aivmx = ["npyz", "base64"]
|
||||
base64 = ["dep:base64"]
|
||||
|
||||
@@ -10,12 +10,23 @@ use std::sync::Arc;
|
||||
|
||||
type JPreprocessType = JPreprocess<DefaultTokenizer>;
|
||||
|
||||
#[cfg(feature = "agpl_dict")]
|
||||
fn agpl_dict() -> Result<Option<UserDictionary>> {
|
||||
Ok(Some(
|
||||
UserDictionary::load(include_bytes!(concat!(env!("OUT_DIR"), "/all.bin")))
|
||||
.map_err(|e| Error::LinderaError(e.to_string()))?,
|
||||
))
|
||||
}
|
||||
|
||||
#[cfg(not(feature = "agpl_dict"))]
|
||||
fn agpl_dict() -> Result<Option<UserDictionary>> {
|
||||
Ok(None)
|
||||
}
|
||||
|
||||
fn initialize_jtalk() -> Result<JPreprocessType> {
|
||||
let sdic =
|
||||
SystemDictionaryConfig::Bundled(kind::JPreprocessDictionaryKind::NaistJdic).load()?;
|
||||
let u = UserDictionary::load(include_bytes!(concat!(env!("OUT_DIR"), "/all.bin")))
|
||||
.map_err(|e| Error::LinderaError(e.to_string()))?;
|
||||
let jpreprocess = JPreprocess::with_dictionaries(sdic, Some(u));
|
||||
let jpreprocess = JPreprocess::with_dictionaries(sdic, agpl_dict()?);
|
||||
Ok(jpreprocess)
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user