feat(ml): ARM NN acceleration

2023-11-04 09:34:19 +01:00
parent 767fe87b2e
commit 5f6ad9e239
4 changed files with 86 additions and 6 deletions
--- a/machine-learning/export/env.yaml
+++ b/machine-learning/export/env.yaml
@@ -20,6 +20,7 @@ dependencies:
  - torchvision
  - transformers==4.*
  - pip:
-    - multilingual-clip
-    - onnx-simplifier
+      - multilingual-clip
+      - onnx-simplifier
+      - tensorflow
 category: main