tflite CLIP export

2023-11-30 22:02:20 +01:00
parent 5f6ad9e239
commit eb0f79b72e
5 changed files with 51 additions and 15 deletions
--- a/machine-learning/pyproject.toml
+++ b/machine-learning/pyproject.toml
@@ -29,6 +29,7 @@ python-multipart = "^0.0.6"
 orjson = "^3.9.5"
 safetensors = "0.3.2"
 gunicorn = "^21.1.0"
+tflite-runtime = "^2.14.0"

 [tool.poetry.group.dev.dependencies]
 mypy = "^1.3.0"