mirror of https://github.com/commaai/tinygrad.git
60 lines
1.4 KiB
Markdown
60 lines
1.4 KiB
Markdown
# tinygrad Showcase
|
|
|
|
Despite being a tiny library, tinygrad is capable of doing a lot of things. From state-of-the-art [vision](https://arxiv.org/abs/1905.11946) to state-of-the-art [language](https://arxiv.org/abs/1706.03762) models.
|
|
|
|
## Vision
|
|
|
|
### EfficientNet
|
|
|
|
You can either pass in the URL of a picture to discover what it is:
|
|
```sh
|
|
python3 examples/efficientnet.py https://media.istockphoto.com/photos/hen-picture-id831791190
|
|
```
|
|
Or, if you have a camera and OpenCV installed, you can detect what is in front of you:
|
|
```sh
|
|
python3 examples/efficientnet.py webcam
|
|
```
|
|
|
|
### YOLOv8
|
|
|
|
Take a look at [yolov8.py](/examples/yolov8.py).
|
|
|
|
|
|
|
|
## Audio
|
|
|
|
### Whisper
|
|
|
|
Take a look at [whisper.py](/examples/whisper.py). You need pyaudio and torchaudio installed.
|
|
|
|
```sh
|
|
SMALL=1 python3 examples/whisper.py
|
|
```
|
|
|
|
## Generative
|
|
|
|
### Generative Adversarial Networks
|
|
|
|
Take a look at [mnist_gan.py](/examples/mnist_gan.py).
|
|
|
|
|
|
|
|
### Stable Diffusion
|
|
|
|
```sh
|
|
python3 examples/stable_diffusion.py
|
|
```
|
|
|
|
|
|
|
|
*"a horse sized cat eating a bagel"*
|
|
|
|
### LLaMA
|
|
|
|
You will need to download and put the weights into the `weights/LLaMA` directory, which may need to be created.
|
|
|
|
Then you can have a chat with Stacy:
|
|
```sh
|
|
python3 examples/llama.py
|
|
```
|