b9dd0bc3b7
## Description - Quantized the CLIP text encoder - Moved preprocessing and postprocessing of face detection inside the model - Optimised the ONNX models more wherever possible - Created a place in infra for ML version control of sorts ## Tests Have tested the changes on mobile, but not on desktop. Please carefully review the changes on desktop, especially regarding the face detection post-processing, more specifically the image (re-)size correction.
35 lines
740 B
TOML
35 lines
740 B
TOML
[project]
|
|
name = "ente-clean-ml"
|
|
version = "0.1.0"
|
|
description = "Add your description here"
|
|
readme = "README.md"
|
|
requires-python = ">=3.12"
|
|
dependencies = [
|
|
"ipykernel>=6.29.5",
|
|
"matplotlib>=3.9.2",
|
|
"numpy>=2.1.2",
|
|
"onnx>=1.17.0",
|
|
"onnxconverter-common>=1.14.0",
|
|
"onnxruntime-extensions>=0.12.0",
|
|
"onnxruntime>=1.19.2",
|
|
"onnxsim>=0.4.36",
|
|
"opencv-python>=4.10.0.84",
|
|
"pandas>=2.2.3",
|
|
"pillow>=10.4.0",
|
|
"pip>=24.2",
|
|
"pyyaml>=6.0.2",
|
|
"requests>=2.32.3",
|
|
"scipy>=1.14.1",
|
|
"seaborn>=0.13.2",
|
|
"thop>=0.1.1.post2209072238",
|
|
"torch>=2.4.1",
|
|
"torchaudio>=2.4.1",
|
|
"torchvision>=0.19.1",
|
|
"tqdm>=4.66.5",
|
|
]
|
|
|
|
[tool.uv]
|
|
dev-dependencies = [
|
|
"ipykernel>=6.29.5",
|
|
]
|