417 Commits

Author SHA1 Message Date
yoni13
dd52c2d68c Update permission 2025-01-19 23:19:44 +08:00
yoni13
20ba9f97e9 update mapping 2025-01-19 13:15:38 +00:00
yoni13
ac4ce3ea9c add input outputs 2025-01-19 12:10:01 +00:00
yoni13
2b967ca358 raise NotImplementedError for now 2025-01-19 12:02:08 +08:00
yoni13
1653cd9cd7 update supported SOCs 2025-01-18 17:38:44 +08:00
yoni13
32f3707e52 fix types and ignored pattern 2025-01-18 17:17:48 +08:00
yoni13
d2b7e10f55 shellcheck happy 2025-01-18 16:56:15 +08:00
yoni13
b3ae5d34cc fix typo in tests 2025-01-18 16:54:18 +08:00
Yoni Yang
4e42fbc091
Merge branch 'main' into rknn-toolkit-lite2 2025-01-18 16:45:38 +08:00
yoni13
9926045d5e add a simple script to notify user if some op is not supported 2025-01-18 16:32:08 +08:00
yoni13
d7381ab5c1 refactor ignore_patterns 2025-01-18 04:48:11 +00:00
yoni13
87a46dcc5e remove unnecessary print 2025-01-18 11:30:49 +08:00
yoni13
be76857ae6 make these functions snake case. 2025-01-18 11:03:54 +08:00
yoni13
f5de3de163 fix typo and add a propper var name 2025-01-18 10:46:57 +08:00
yoni13
3634ae1f5b fix granularity 2025-01-18 09:58:39 +08:00
Tempest
d5a9294eeb
feat: Add additional env variables to ML container (#15398)
* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Update config.py

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Apply formatting

* minor update

* formatting

* root validator

* minor update

* minor update

* minor update

* change to support explicit models

* minor update

* minor change

* minor change

* minor change

* minor update

* add logs, resolve errors

* minor change

* add new enviornment variables

* minor revisons

* remove comments

* add additional variables to ML (fixed)

* add additional variables to ML (fixed)

* add additional variables to ML

* formatting

* remove comment

* remove mypy error

* remove unused module

* merge f strings
2025-01-17 17:22:05 -05:00
yoni13
f067212491 tpe 2025-01-17 19:56:23 +08:00
yoni13
bc48b67379 switch to sha256 2025-01-17 19:42:23 +08:00
yoni13
26d5fb0ac6 add checksum for libnnrt.so 2025-01-17 19:39:26 +08:00
yoni13
f32d991131 changes some cases 2025-01-17 19:25:01 +08:00
Tempest
c5476a99b1
feat: Add additional env variables for Machine Learning (#15326)
* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Update config.py

* Add additional variables to preload part ML models

* Add additional variables to preload part ML models

* Apply formatting

* minor update

* formatting

* root validator

* minor update

* minor update

* minor update

* change to support explicit models

* minor update

* minor change

* minor change

* minor change

* minor update

* add logs, resolve errors

* minor change

* add new enviornment variables

* minor revisons

* remove comments
2025-01-14 22:06:01 +00:00
renovate[bot]
073fccb517
chore(deps): update python:3.11-slim-bookworm docker digest to 6ed5bff (#15346)
Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>
2025-01-14 18:33:27 +00:00
yoni13
9882b83cd4 Should FIx the quote that made mypy unhappy 2025-01-15 00:35:43 +08:00
renovate[bot]
19e2504583
fix(deps): update machine-learning (#15336) 2025-01-14 10:19:01 -05:00
yoni13
b5a4ed5160 this duplicated? 2025-01-14 19:22:28 +08:00
yoni13
0f03f77e8e remove non implemented tests 2025-01-14 19:02:16 +08:00
yoni13
c21ce40d9c switch to Runtime error instead of exit() 2025-01-14 18:50:21 +08:00
yoni13
5244ed6d4d black app export 2025-01-14 18:40:28 +08:00
yoni13
8b80d034cb fixed some bugs 2025-01-14 10:38:45 +00:00
yoni13
4b0f93cf6a add test,founds bugs, fix it tomorrow 2025-01-14 01:08:44 +08:00
yoni13
6c4e6cb96f reformat 2025-01-13 18:37:25 +08:00
yoni13
b6cc2054c5 ignore rknn model if not using it 2025-01-13 18:37:01 +08:00
yoni13
2f7e44aa63 typing be happy. 2025-01-13 18:24:12 +08:00
yoni13
ebdfe1b7b6 Load model by SOC name 2025-01-13 17:08:16 +08:00
yoni13
daf886088a Add export script 2025-01-13 05:44:22 +00:00
yoni13
4c7ac1438b only load knnx model when required 2025-01-12 19:11:16 +00:00
Yoni Yang
8965a9fb16
Merge branch 'main' into rknn-toolkit-lite2 2025-01-13 01:41:55 +08:00
yoni13
e6ff21b345 set default thread num to 2, not everyone has 8 gigs of ram 2025-01-12 16:05:18 +08:00
renovate[bot]
581d32269d
fix(deps): update machine-learning (#15247)
Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>
2025-01-11 22:05:20 -06:00
yoni13
c665fd2625 Fix Please do not set this parameter on other platforms. 2025-01-12 01:29:24 +08:00
yoni13
c72cf61ed0 support core_mask for specfic socs 2025-01-12 01:24:24 +08:00
yoni13
19ee48f6f0 fix path 2025-01-12 01:09:36 +08:00
yoni13
efaf70eb9d Set running threads from env 2025-01-12 01:02:16 +08:00
yoni13
665718b09e add rknn to src 2025-01-11 21:11:30 +08:00
yoni13
807111e3b5 Should Fix No module named 'rknn' 2025-01-11 20:41:38 +08:00
yoni13
815ed1ae66 Install onnxruntime 2025-01-11 20:33:48 +08:00
yoni13
416211916d Check if NPU drivers is loaded or not. 2025-01-11 17:59:16 +08:00
yoni13
23d0ea0e7b ruff 2025-01-11 16:28:26 +08:00
yoni13
d5e453a773 ruff format 2025-01-11 16:26:17 +08:00
yoni13
7f2af6f819 Fix typo: rknnlite.api 2025-01-11 16:19:51 +08:00