chore: split extension and wasm tasks
This commit is contained in:
parent
4152af652b
commit
2109aaa16d
1 changed files with 18 additions and 50 deletions
|
|
@ -6,38 +6,37 @@ _MK_WASMPACK_MOD = "--release"
|
|||
|
||||
[tasks.release]
|
||||
dependencies = [
|
||||
"build_inference_models",
|
||||
"build_extension"
|
||||
"models",
|
||||
"extension"
|
||||
]
|
||||
|
||||
[tasks.dev]
|
||||
dependencies = [
|
||||
"build_inference_models",
|
||||
"extension_npm_setup"
|
||||
]
|
||||
cwd = "extension"
|
||||
command = "pnpm"
|
||||
args = ["dev"]
|
||||
|
||||
[tasks.build_extension]
|
||||
[tasks.extension]
|
||||
dependencies = [
|
||||
"build_extension_mozf",
|
||||
"zip_extension_mozf",
|
||||
"build_extension_chro",
|
||||
"zip_extension_chro"
|
||||
"extension_mozf",
|
||||
"extension_mozf_zip",
|
||||
"extension_chro",
|
||||
"extension_chro_zip"
|
||||
]
|
||||
|
||||
[tasks.zip_extension_chro]
|
||||
[tasks.extension_chro_zip]
|
||||
cwd = "extension"
|
||||
command = "pnpm"
|
||||
args = ["zip:chrome"]
|
||||
|
||||
[tasks.zip_extension_mozf]
|
||||
[tasks.extension_mozf_zip]
|
||||
cwd = "extension"
|
||||
command = "pnpm"
|
||||
args = ["zip:firefox"]
|
||||
|
||||
[tasks.build_extension_chro]
|
||||
[tasks.extension_chro]
|
||||
dependencies = [
|
||||
"extension_npm_setup"
|
||||
]
|
||||
|
|
@ -45,7 +44,7 @@ cwd = "extension"
|
|||
command = "pnpm"
|
||||
args = ["build:chrome"]
|
||||
|
||||
[tasks.build_extension_mozf]
|
||||
[tasks.extension_mozf]
|
||||
dependencies = [
|
||||
"extension_npm_setup"
|
||||
]
|
||||
|
|
@ -58,13 +57,13 @@ cwd = "extension"
|
|||
command = "pnpm"
|
||||
args = ["install"]
|
||||
|
||||
[tasks.build_inference_models]
|
||||
[tasks.models]
|
||||
dependencies = [
|
||||
"build_inference_models_ai_text",
|
||||
"build_inference_models_ai_text_lite"
|
||||
"inference_model_ai_text",
|
||||
"inference_model_ai_text_lite"
|
||||
]
|
||||
|
||||
[tasks.build_inference_models_ai_text]
|
||||
[tasks.inference_model_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"
|
||||
|
|
@ -76,10 +75,11 @@ args = [
|
|||
"--target", "web",
|
||||
"--out-dir", "pkg/inference-ai-text",
|
||||
"--out-name", "inference-ai-text",
|
||||
"--features", "ai_text"
|
||||
"--features", "ai_text",
|
||||
"-vvv"
|
||||
]
|
||||
|
||||
[tasks.build_inference_models_ai_text_lite]
|
||||
[tasks.inference_model_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"
|
||||
|
|
@ -93,35 +93,3 @@ args = [
|
|||
"--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"]
|
||||
|
|
|
|||
Loading…
Reference in a new issue