127 lines
2.8 KiB
TOML
127 lines
2.8 KiB
TOML
[env.development]
|
|
_MK_WASMPACK_MOD = "--dev"
|
|
|
|
[env.production]
|
|
_MK_WASMPACK_MOD = "--release"
|
|
|
|
[tasks.release]
|
|
dependencies = [
|
|
"build_inference_models",
|
|
"build_extension"
|
|
]
|
|
|
|
[tasks.dev]
|
|
dependencies = [
|
|
"build_inference_models",
|
|
"extension_npm_setup"
|
|
]
|
|
cwd = "extension"
|
|
command = "pnpm"
|
|
args = ["dev"]
|
|
|
|
[tasks.build_extension]
|
|
dependencies = [
|
|
"build_extension_mozf",
|
|
"zip_extension_mozf",
|
|
"build_extension_chro",
|
|
"zip_extension_chro"
|
|
]
|
|
|
|
[tasks.zip_extension_chro]
|
|
cwd = "extension"
|
|
command = "pnpm"
|
|
args = ["zip:chrome"]
|
|
|
|
[tasks.zip_extension_mozf]
|
|
cwd = "extension"
|
|
command = "pnpm"
|
|
args = ["zip:firefox"]
|
|
|
|
[tasks.build_extension_chro]
|
|
dependencies = [
|
|
"extension_npm_setup"
|
|
]
|
|
cwd = "extension"
|
|
command = "pnpm"
|
|
args = ["build:chrome"]
|
|
|
|
[tasks.build_extension_mozf]
|
|
dependencies = [
|
|
"extension_npm_setup"
|
|
]
|
|
cwd = "extension"
|
|
command = "pnpm"
|
|
args = ["build:firefox"]
|
|
|
|
[tasks.extension_npm_setup]
|
|
cwd = "extension"
|
|
command = "pnpm"
|
|
args = ["install"]
|
|
|
|
[tasks.build_inference_models]
|
|
dependencies = [
|
|
"build_inference_models_ai_text",
|
|
"build_inference_models_ai_text_lite"
|
|
]
|
|
|
|
[tasks.build_inference_models_ai_text]
|
|
env = {_MK_MODEL_AI_TEXT = "onnx-community/modernbert-ai-detection-raid-mage-ONNX",_MK_MODEL_AI_TEXT_ATTENTION = "yes",_MK_MODEL_AI_TEXT_TYPES = "no"}
|
|
cwd = "wasm/inference-models"
|
|
install_crate = "wasm-pack"
|
|
command = "wasm-pack"
|
|
args = [
|
|
"build",
|
|
"${_MK_WASMPACK_MOD}",
|
|
"--reference-types",
|
|
"--target", "web",
|
|
"--out-dir", "pkg/inference-ai-text",
|
|
"--out-name", "inference-ai-text",
|
|
"--features", "ai_text"
|
|
]
|
|
|
|
[tasks.build_inference_models_ai_text_lite]
|
|
env = {_MK_MODEL_AI_TEXT = "onnx-community/e5-small-lora-ai-generated-detector-ONNX",_MK_MODEL_AI_TEXT_ATTENTION = "yes",_MK_MODEL_AI_TEXT_TYPES = "yes"}
|
|
cwd = "wasm/inference-models"
|
|
install_crate = "wasm-pack"
|
|
command = "wasm-pack"
|
|
args = [
|
|
"build",
|
|
"${_MK_WASMPACK_MOD}",
|
|
"--reference-types",
|
|
"--target", "web",
|
|
"--out-dir", "pkg/inference-ai-text-lite",
|
|
"--out-name", "inference-ai-text-lite",
|
|
"--features", "ai_text"
|
|
]
|
|
|
|
[tasks.export_onnx]
|
|
dependencies = ["export_onnx_ai_text"]
|
|
|
|
[tasks.export_onnx_ai_text]
|
|
dependencies = ["exportenv"]
|
|
command = ".export_venv/bin/optimum-cli"
|
|
args = [
|
|
"export", "onnx",
|
|
"--model", "GeorgeDrayson/modernbert-ai-detection-raid-mage",
|
|
"--task", "text-classification",
|
|
"--dtype", "fp32",
|
|
"--opset", "18",
|
|
"--optimize", "O1",
|
|
"--monolith",
|
|
"--no-post-process",
|
|
"wasm/inference-models/models/ai_text"
|
|
]
|
|
|
|
[tasks.exportenv]
|
|
dependencies = [
|
|
"exportenv_create",
|
|
"exportenv_setup"
|
|
]
|
|
|
|
[tasks.exportenv_create]
|
|
command = "python3"
|
|
args = ["-m", "venv", ".export_venv"]
|
|
|
|
[tasks.exportenv_setup]
|
|
command = ".export_venv/bin/pip"
|
|
args = ["install", "--upgrade", "-r", "exportenv_requirements.txt"]
|