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