461 Commits

Author	SHA1	Message	Date
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
yoni13	f4671f4886	Indentation issue	2025-01-11 16:16:26 +08:00
yoni13	7aaf3aa57b	Remove unused imports.	2025-01-11 16:03:33 +08:00
yoni13	506ca0d3a4	Dockerfile for rknn	2025-01-11 15:47:24 +08:00
yoni13	d5ef821b24	Set group RKNN to optional	2025-01-11 15:30:05 +08:00
yoni13	d10147f478	Handling Import and file not found Error for non-arm devices.	2025-01-11 15:19:53 +08:00
yoni13	c20d110257	support for rknn.rknnpool.is_available	2025-01-11 10:39:45 +08:00