Commit Graph

32 Commits

Author SHA1 Message Date
Abhishek Varma
c3c701e654 Update requirements.txt + README.md of SD
-- This commit includes two python modules as part of requirements.txt.
-- It also updates README.md to also inclue `--no-use_tuned` for users to
   be able to try `hf_model_id` or `ckpt_loc` without any issue.

Signed-off-by: Abhishek Varma <abhishek@nod-labs.com>
2023-01-30 14:12:54 +05:30
Ean Garvey
fb6beae27c Adds pytest-forked dependency to fix pytest memory accumulation issues. (#876)
* Minor improvements to test-models workflow

- cleaned up pytest command line args in Validate Models job scripts.
- Removed -s flag to provide more readable logs
- Changed shark_cache location to within github workspace and removed --update_tank flag from Linux workflows.

* Use pytest-forked for managing pytest memory usage.
2023-01-26 18:20:15 -06:00
powderluv
03543030d3 use pefile 2023-01-24 18:35:51 -08:00
powderluv
fc6b474b92 Add ordlookup to requirements.txt 2023-01-24 18:30:16 -08:00
powderluv
ecfdec12f3 Update requirements.txt 2022-12-25 15:39:20 -08:00
Gaurav Shukla
b03038222d [SHARK] Update dependencies
Signed-Off-by: Gaurav Shukla <gaurav@nod-labs.com>
2022-12-13 22:12:00 +05:30
Daniel Garvey
8859853b41 Revert "Revert "find gsutil on linux (#557)" (#560)" (#561)
This reverts commit 3c46021102.
2022-12-05 20:27:43 -08:00
Daniel Garvey
3c46021102 Revert "find gsutil on linux (#557)" (#560)
This reverts commit bba8646669.
2022-12-05 21:53:47 -06:00
Daniel Garvey
bba8646669 find gsutil on linux (#557)
* find gsutil on linux

* cleaned up downloader and ditched gsutil

Co-authored-by: dan <dan@nod-labs.com>
2022-12-05 19:03:48 -08:00
powderluv
83b138208d Add gradio to requirements.txt 2022-12-03 16:06:52 -08:00
Anush Elangovan
1106456651 Update cuda 11.7 nightly URL and add index.spec 2022-12-01 22:49:23 -08:00
powderluv
d5ac841a1a Update requirements.txt
add transformers to base venv
2022-11-30 12:12:28 -08:00
Ean Garvey
9035a2eed3 Add --local_tank_cache flag and update requirements. (#368)
* Add --local_tank_cache flag and update requirements.

* Update requirements-importer.txt
2022-09-28 03:02:59 -05:00
Phaneesh Barwaria
1322ec5935 Simplified Testing Interface (#289) 2022-08-24 23:54:56 -05:00
Chi_Liu
17dba601c8 Add huggingface top5 image classification automodel (#268) 2022-08-22 15:05:38 -07:00
Ean Garvey
0ee515a7be Fix GPU benchmarks on PyTorch model tests. (#218)
* Add CUDA_BENCHMARKS option to setup.venv to enable PyTorch benchmarks on CUDA.

* Fix PyTorch GPU benchmarks for tank models.
2022-07-29 15:17:55 -07:00
Ean Garvey
c5ae01311d Make PyTorch model tests importer-independent and add benchmarks. 2022-07-15 21:54:41 -05:00
monorimet
b647764b9a Update requirements.txt 2022-07-14 13:07:46 -05:00
monorimet
08448a013f Refactor TF tests for importer split, update pytorch tests. 2022-07-14 12:59:21 -05:00
powderluv
be66cc6ac1 Update requirements.txt 2022-06-12 01:06:47 -07:00
Anush Elangovan
3c26c02a5e Add OSX-importer requirements 2022-06-12 07:38:02 +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
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
Stanley Winata
8565be9b6b ORT-HF Benchmark Integration (#101)
-Add HF Benchmarker class.
-Add sample to benchmark HF model.

Example:
```bash
python -m benchmarks.hf_model_benchmark --num_iterations=10 --model_name="microsoft/MiniLM-L12-H384-uncased"
```
2022-06-07 23:49:39 -07:00
powderluv
8dd6b8cbd1 Update requirements.txt 2022-06-04 15:51:38 -07:00
anush elangovan
81881a36ea Fix accidental additions 2022-06-03 06:43:08 +00:00
anush elangovan
8fa5b39bd6 Add importer utils to linux build 2022-06-02 19:33:51 +00:00
Prashant Kumar
dcc1bbac14 Add pytests-parametrize with device and dynamic option.
Added pytests-parametrize with device and dynamic options, also
added worker flag in pytest to run tests concurrently.
2022-05-17 11:07:56 +05:30
Prashant Kumar
bd212634c1 Add CI for testing of various models. (#28)
Added pytest CI for testing for Pytorch Inference and vision models.
2022-04-27 06:52:49 -07:00
Prashant Kumar
0502c1b384 Add necessary packages installation in install.sh.
I can't find how to maintain order in requirements.txt and
hence install.sh solves this issue.
2022-04-14 22:05:51 +05:30
Prashant Kumar
59485f571e Update readme.md of shark_runner
Updated the readme.md of shark_runner and added setup.py for pip
installation.
2022-04-01 01:00:32 +05:30