Commit Graph

254 Commits

Author SHA1 Message Date
stanley
14a56ca9b0 Mini LM Loader Example
-Add example to load miniLM from SharkHUB and benchmark.
-Modify TF benchmark to have growing GPU allocation.
-Add shark_load helper function
2022-06-15 02:57:42 +00:00
Stanley Winata
95c2e3d6ea Enable Benchmark tests (#129) 2022-06-14 14:05:26 -07:00
powderluv
cb7007ce16 Update test-models.yml
remove pip cache hang on osx and enable benchmarks on perf-linux
2022-06-14 12:23:07 -07:00
Prashant Kumar
380b0b7c54 Add seq_classification shark module to run seq_classification
on hugging face supported models.
2022-06-14 22:36:19 +05:30
Ean Garvey
01f900a9ff Move conftest.py to fix nightly release. (#126)
* Move conftest.py into frontend directory.

* xfail albert static on vulkan backend.
2022-06-14 07:07:33 -07:00
Stanley Winata
e48642f8a5 Update benchmark command to ToM and Add to test (#125)
-Update benchmark_cl to latest benchmark_module API
-tensorflow OOM issue.
2022-06-13 23:57:05 -07:00
Ean Garvey
61b19ce091 Merge pull request #122 from nod-ai/ean-tests
Enable passing of --save-mlir cmd-line option through pytest.
2022-06-13 20:27:20 -05:00
Ean Garvey
ec13200c7c Update conftest.py 2022-06-13 19:51:21 -05:00
Ean Garvey
e872296498 Enable passing of --save-mlir cmd-line option through pytest. 2022-06-14 00:27:34 +00:00
Anush Elangovan
d81d5a35df pin numpy for onnx 2022-06-13 21:28:26 +00:00
Anush Elangovan
c6de60dab5 Disable demote 2022-06-13 16:55:28 +00:00
powderluv
3cbd697aa6 Update test-models.yml 2022-06-13 09:12:26 -07:00
Ean Garvey
acdd1d1785 Update model testing utilities. (#119) 2022-06-13 02:16:30 -07:00
Anush Elangovan
5991d80c1b Pin transformers==4.18.0 2022-06-13 05:25:13 +00:00
Anush Elangovan
3ea34d08e2 Add __init__.py for tank/ 2022-06-12 21:53:59 -07:00
Ean Garvey
cbd131d588 Refactor testing framework. (#117) 2022-06-12 20:15:30 -07:00
powderluv
eb3781ddb2 Update README.md 2022-06-12 17:31:24 -07:00
powderluv
bcf9dbb3bf Update test-models.yml 2022-06-12 17:28:18 -07:00
powderluv
ac84f3b0d5 Update test-models.yml 2022-06-12 17:08:00 -07:00
powderluv
719d053ba8 Update test-models.yml 2022-06-12 17:03:18 -07:00
powderluv
c68e572f1a Update test-models.yml
add a100
2022-06-12 15:55:39 -07:00
powderluv
f77c2aa54a Update nightly.yml 2022-06-12 14:53:10 -07:00
powderluv
579ba897ad Update requirements-importer.txt 2022-06-12 14:16:09 -07:00
powderluv
37592d9075 Update requirements-importer.txt 2022-06-12 11:19:01 -07:00
powderluv
4ed8e8b54d Update nightly.yml 2022-06-12 09:51:30 -07:00
powderluv
87f5da7a9a Update nightly.yml 2022-06-12 09:34:26 -07:00
powderluv
07f93ef481 Update nightly.yml 2022-06-12 09:25:48 -07:00
powderluv
2aa02e0fed Update nightly.yml 2022-06-12 09:19:25 -07:00
powderluv
b0fb0ed5b8 Update nightly.yml 2022-06-12 09:08:46 -07:00
powderluv
96088d6d34 Update nightly.yml 2022-06-12 08:41:05 -07:00
powderluv
d9ecb3e964 Update nightly.yml 2022-06-12 08:21:44 -07:00
Anush Elangovan
e4dd179a73 Tweak requirements.txt 2022-06-12 02:01:29 -07:00
Anush Elangovan
879942d87f Fix macOS importer reqs 2022-06-12 01:45:05 -07:00
powderluv
be66cc6ac1 Update requirements.txt 2022-06-12 01:06:47 -07:00
powderluv
d6785cdce7 Update test-models.yml 2022-06-12 01:02:03 -07:00
powderluv
c2e06b9a9b Update requirements-importer.txt 2022-06-12 00:47:59 -07:00
powderluv
96c3e9e387 Update requirements-importer-macos.txt 2022-06-12 00:47:36 -07:00
powderluv
4b35dec83b Update test-models.yml 2022-06-12 00:43:22 -07:00
Anush Elangovan
3c26c02a5e Add OSX-importer requirements 2022-06-12 07:38:02 +00:00
Anush Elangovan
cd8fe34baa Fix local-task vs dylib naming changes 2022-06-12 07:21:48 +00:00
powderluv
c0be20d835 Update requirements.txt
point to latest nightly cpu
2022-06-12 00:05:48 -07:00
powderluv
fb9ded8b09 Update requirements.txt
re-add torch / torchvision in main requirements.txt
2022-06-12 00:05:27 -07:00
powderluv
35fa22f167 Update test-models.yml 2022-06-11 00:54:27 -07:00
powderluv
9043dcedc7 Update nightly.yml 2022-06-11 00:54:02 -07:00
powderluv
1887e8e3ef Update test-models.yml 2022-06-11 00:42:07 -07:00
powderluv
61b260b6cf Update nightly.yml 2022-06-11 00:41:40 -07:00
Anush Elangovan
626482e46a Refactor setup_venv.sh
IMPORTER=1 ./setup_venv.sh is required to get importer deps

Without importers venv is 1.1G, with it is 3.9G
2022-06-11 06:26:31 +00:00
Prashant Kumar
93bf13199f Add tensorflow models under AutoModelForMaskedLM class. 2022-06-09 22:02:21 +05:30
Chi_Liu
8640c3ebfe Add shark_importer tflite module and albert_shark_test example (#110) 2022-06-09 06:17:22 -07:00
Prashant Kumar
bf02c8d0db Add t5(https://huggingface.co/docs/transformers/model_doc/t5) model. 2022-06-08 16:25:23 +00:00