tinygrad/docs/showcase.md

1.6 KiB

Showcase

Despite being a tiny library, tinygrad is capable of doing a lot of things. From state-of-the-art vision to state-of-the-art language models.

Vision

EfficientNet

You can either pass in the URL of a picture to discover what it is:

python3 examples/efficientnet.py ./test/models/efficientnet/Chicken.jpg

Or, if you have a camera and OpenCV installed, you can detect what is in front of you:

python3 examples/efficientnet.py webcam

YOLOv8

Take a look at yolov8.py.

yolov8 by tinygrad

Audio

Whisper

Take a look at whisper.py. You need pyaudio and torchaudio installed.

SMALL=1 python3 examples/whisper.py

Generative

Stable Diffusion

python3 examples/stable_diffusion.py

a horse sized cat eating a bagel

"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:

python3 examples/llama.py

Conversation

Make sure you have espeak installed and PHONEMIZER_ESPEAK_LIBRARY set.

Then you can talk to Stacy:

python3 examples/conversation.py