github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-09 13:57:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

Update README.md with turbine compilable models

Browse Source
This commit is contained in:
Daniel Garvey
2024-01-16 09:08:43 -06:00
committed by GitHub
parent fa95ed30d1
commit 5ebafec1e2
1 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
README.md
View File

@@ -334,15 +334,19 @@ result = shark_module.forward((arg0, arg1))
SHARK is maintained to support the latest innovations in ML Models:
| TF HuggingFace Models | SHARK-CPU | SHARK-CUDA | SHARK-METAL |
|---------------------|----------|----------|-------------|
| BERT | :green_heart: | :green_heart: | :green_heart: |
| DistilBERT | :green_heart: | :green_heart: | :green_heart: |
| GPT2 | :green_heart: | :green_heart: | :green_heart: |
| BLOOM | :green_heart: | :green_heart: | :green_heart: |
| Stable Diffusion | :green_heart: | :green_heart: | :green_heart: |
| Vision Transformer | :green_heart: | :green_heart: | :green_heart: |
| ResNet50 | :green_heart: | :green_heart: | :green_heart: |
| hf_model_id | SHARK-Turbine |
|---------------------|----------|
| microsoft/resnet-50 | :green_heart: |
| microsoft/resnet-18 | :green_heart: |
| microsoft/resnet-101 | :green_heart: |
| microsoft/resnet-152 | :green_heart: |
| meta-llama/Llama-2-7b-chat-hf | :green_heart: |
| chavinlo/alpaca-native | :green_heart: |
| CompVis/stable-diffusion-v1-4 | :green_heart: |
| runwayml/stable-diffusion-v1-5 | :green_heart: |
| stablediffusionapi/anything-v5 | :green_heart: |
For a complete list of the models supported in SHARK, please refer to [tank/README.md](https://github.com/nod-ai/SHARK/blob/main/tank/README.md).