DUB-AI: multi-task модель (тип озвучки + автор-спан + 18+), обучение/инференс/ONNX/сервис, тесты и документация

This commit is contained in:
DUB-AI committed 2026-10-27 16:43:36 +07:00
commit 8dbca3f4a3
2873 files changed
+44479

No files matched your search

+27
View File
@@ -0,0 +1,27 @@
# onnx_local — локальный инференс DUB-AI на Rust (без Python)
```bash
cargo build
# onnxruntime.dll грузится динамически; можно взять из pip-пакета onnxruntime
# (уже есть в .venv проекта) или из официального релиза onnxruntime
ORT_DYLIB_PATH=<путь>/onnxruntime.dll ./target/debug/dub-ai-onnx-local.exe \
"<путь к models/best-onnx>" "Перевод: Дублированный | MovieDalen"
```
Пример из корня проекта (Git Bash):
```bash
cd examples/rust/onnx_local
ORT_DYLIB_PATH="$(find ../../.venv -name onnxruntime.dll | head -1)" \
./target/debug/dub-ai-onnx-local.exe ../../models/best-onnx \
"Авторский (одноголосый закадровый) - Юрий Сербин"
```
Модель должна быть экспортирована заранее: `python -m src.export_onnx`
(создаст `models/best-onnx/`: model.onnx, tokenizer.json, label_maps.json,
aliases.json).
Крейты: `ort 2.0.0-rc.15` (инференс), `tokenizers` (HF tokenizer.json),
`serde_json` (метки/алиасы). Постобработка повторяет `src/predict.py`:
argmax типа и 18+, первый B-I-прогон по offsets → текст спана →
канонизация по aliases.json.